python - 使用 NumPy 数组交换列

Question

当我有 a=1 和 b=2 时，我可以写 a,b=b,a 这样 a 和 b 可以互换。

我将这个矩阵用作数组:

   [ 1,  2,  0, -2]
   [ 0,  0,  1,  2]
   [ 0,  0,  0,  0]

交换 numpy 数组的列不起作用:

import numpy as np

x = np.array([[ 1,  2,  0, -2],
   [ 0,  0,  1,  2],
   [ 0,  0,  0,  0]])

x[:,1], x[:,2] = x[:,2], x[:,1]

它产生:

   [ 1,  0,  0, -2]
   [ 0,  1,  1,  2]
   [ 0,  0,  0,  0]

所以 x[:,1] 只是被覆盖了，并没有转移到 x[:,2]。

为什么会这样？

Answer 1

如果你想交换列，你可以这样做

print x
x[:,[2,1]] = x[:,[1,2]]
print x

输出

[[ 1  2  0 -2]
 [ 0  0  1  2]
 [ 0  0  0  0]]
[[ 1  0  2 -2]
 [ 0  1  0  2]
 [ 0  0  0  0]]

你在问题​​中提到的交换方法似乎适用于一维数组和列表，

x =  np.array([1,2,0,-2])
print x
x[2], x[1] = x[1], x[2]
print x

输出

[ 1  2  0 -2] 
[ 1  0  2 -2]