NotImplementedError：range_state_int64 不能表示为 Numpy dtype

Question

我正在尝试编写一个函数来初始化一个数组并在返回之前对其进行洗牌。 将 numba 导入为 nb

@nb.jit(nopython=True, cache=True)
def test(x):
    ind = np.array(range(len(x)))
    np.random.shuffle(ind)
    return ind

错误消息说我使用了不支持的功能或数据类型：

NotImplementedError: range_state_int64 cannot be represented as a Numpy dtype

numba 是否支持 numpy.random.shuffle()？如何修改它？谢谢！

Answer 1

这其实和random.shuffle无关，因为numba supports the random module out of the box。

这里的问题是numba cannot support the range object（因为它是一个python对象）当nopython标志被设置时。将range 替换为np.arange：

@nb.njit(cache=True)  # same as @nb.jit(nopython=True, ...)
def test(x):
    ind = np.arange(len(x))
    np.random.shuffle(ind)
    return ind

test([1, 2, 3])
# array([1, 0, 2])