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