python numba : how to slice a column from a numpy array?

Question

我知道 numba 在 nopython 模式下支持许多 numpy 函数: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