Pythonic方式访问numpy数组的移位版本？

Question

什么是pythonic方式来访问numpy数组的移位（左右）？一个明显的例子：

a = np.array([1.0, 2.0, 3.0, 4.0])

是否可以访问：

a_shifted_1_left = np.array([2.0, 3.0, 4.0, 1.0])

来自numpy库？

Answer 1

您正在寻找np.roll -

np.roll(a,-1) # shifted left
np.roll(a,1) # shifted right

示例运行 -

In [28]: a
Out[28]: array([ 1.,  2.,  3.,  4.])

In [29]: np.roll(a,-1) # shifted left
Out[29]: array([ 2.,  3.,  4.,  1.])

In [30]: np.roll(a,1) # shifted right
Out[30]: array([ 4.,  1.,  2.,  3.])

如果您想要更多班次，请转到np.roll(a,-2)和np.roll(a,2)，依此类推。