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

1mrurvl1  于 2023-04-30  发布在  其他
关注(0)|答案(1)|浏览(95)

假设我有一个像这样的二维numpy数组

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

然后我想把它转换成

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

这可能是一个可变的列数,我想将第一列添加到每隔一列,然后再删除它。

uubf1zoe

uubf1zoe1#

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

这个…是用来处理N维数组的
Means -〉将[0]列与第一列之后的所有列相加。
如果你只使用一个矩阵,你可以用途:

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

相关问题