Python、Numpy：如何一次更改兩個向量

Question

我正在嘗試一次更改 NumPy 矩陣的兩個向量，但我丟失了一個向量分量：

import numpy as np

data = np.array([[1, 2, 3, 4],[1, 2, 3, 4],[1, 2, 3, 4],[1, 2, 3, 4]])
last = data[:, -1]
print(last)
data[:, 1:] = data[:, :-1]
data[:, 0] = last

print(data)

給出這個結果：

[4 4 4 4]
[[3 1 2 3]
 [3 1 2 3]
 [3 1 2 3]
 [3 1 2 3]]

但我想保持第一列中的4s。 有什么形式可以實現嗎？

Answer 1

利用：

>>> np.roll(data, 1, 1)
array([[4, 1, 2, 3],
       [4, 1, 2, 3],
       [4, 1, 2, 3],
       [4, 1, 2, 3]])