python - 如何将 numpy 矩阵的列移动到最后一列？

Question

假设我有一个这样的 numpy 矩阵:

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

我想将矩阵的第 2 列移到最后一列:

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

我该怎么做？

Answer 1

使用numpy.roll:

arr[:, 1:] = np.roll(arr[:, 1:], -1, 1)

输出:

array([[1, 4, 7, 8, 3],
       [5, 8, 2, 6, 6],
       [2, 3, 3, 6, 9],
       [7, 9, 3, 5, 1]])

如何:

np.roll 接受三个参数:a、shift 和 axis:

np.roll(a = arr[:, 1:], shift = -1, axis = 1)

这意味着，取 arr[:, 1:](所有行，所有列从 1 开始)，并将其向左移动一个单位(-1。向右将是 + 1), 沿轴1(即柱状移位，轴0为行移位)。

np.roll，顾名思义，是一个循环移位。一个单位偏移将使最后一列成为第一列，依此类推。