包装一个 2D numpy 数组
Wrapping a 2D numpy array
假设我有以下 numpy 二维数组，其中 arr[2,0] = 7.
arr =
[[1 2 3]
[4 5 6]
[7 8 9]]
我怎样才能实现 包装效果 以便当我尝试检索 arr[3,0] 时，我不会得到 "索引 3 超出轴 0 的范围，大小为 3"，但会在 return?
中得到 arr[0,0] = 1
谢谢！
您可以使用 take:
一次索引一个维度
In [72]: np.take(arr,3,0, mode='wrap')
Out[72]: array([1, 2, 3])
假设我有以下 numpy 二维数组，其中 arr[2,0] = 7.
arr =
[[1 2 3]
[4 5 6]
[7 8 9]]
我怎样才能实现 包装效果 以便当我尝试检索 arr[3,0] 时，我不会得到 "索引 3 超出轴 0 的范围，大小为 3"，但会在 return?
arr[0,0] = 1
谢谢！
您可以使用 take:
In [72]: np.take(arr,3,0, mode='wrap')
Out[72]: array([1, 2, 3])