Python 何在numpy中進行循環移位

woff

我有一個numpy數組，例如

a = np.arange(10)
如何將第一個n元素移動到數組的末尾？

我找到了這個roll函數，但似乎只是反過來，它將最後n個元素移到了開頭。

2 個答案:
答案0 :(得分：37)

為什麼不只是roll帶有負數？

1. import numpy as np
   
2. a = np.arange(10)
   
3. np.roll(a,2)
   
4. array([8, 9, 0, 1, 2, 3, 4, 5, 6, 7])
   
5. np.roll(a,-2)
   
6. array([2, 3, 4, 5, 6, 7, 8, 9, 0, 1])

複製代碼

你可以使用負移位

1. a = np.arange(10)
   
2. print(np.roll(a, 3))
   
3. print(np.roll(a, -3))

複製代碼

返回
[7,8,9,0,1,2,3,4,5,6]
[3,4,5,6,7,8,9,0,1,2]