Numpy 将一列添加到其他列并删除

Question

假设我有一个像这样的 2d numpy 数组

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

然后我想把它转换成

[[3,4],
 [9,10],
 [15,16]]

这可能是可变数量的列，我想将第一列添加到其他列中，然后再将其删除。

Answer 1

a = a[..., [0]] + a[..., 1:]

...是为了与 N 维数组一起工作

方法 -> 将 [0] 列与第一列之后的所有列相加。

如果你只使用一个矩阵，你可以使用：

a = a[:, [0]] + a[:, 1:]