python numba：如何从numpy数组中切出一列？

Question

我明白numba支持许多numpy的功能nopython模式：http://numba.pydata.org/numba-doc/dev/reference/numpysupported.html

然而，有一个方法切片从nopython模式numpy的阵列的列或行的？

喜欢的东西

x= y[:,mycol] 
x=y[myrow,:] 

不nopython模式下工作。我发现的唯一解决方法是将数组切片到numba外部，并将'切片'传递给我的numba函数，但这并不总是可行或简单。

谢谢！

Answer 1

你使用的是哪种版本的Numba？使用0.26.0，以下工作：

import numpy as np 
import numba as nb 

@nb.jit(nopython=True) 
def func(x, mycol): 
    y = x[:,mycol] 

    return y 

x = np.arange(100).reshape((10,10)) 
mycol = 2 
y = func(x, mycol) 

print np.allclose(x[:,mycol], y) # True