python+numpy交换数组的两行或者两列

其实很简单：

1. 交换两行：

import numpy as np
arr=np.arange(16).reshape(4,4)
# 结果如下
>>>array([[ 0,  1,  2,  3],
       [ 4,  5,  6,  7],
       [ 8,  9, 10, 11],
       [12, 13, 14, 15]])

# [0,1]表示为第一行、第二行，逗号前是表示行的参数，逗号后的表示列，冒号表示所有列
arr[[0,1],:]=arr[[1,0],:]
print(arr)
# 结果如下
>>>array([[ 4,  5,  6,  7],
       [ 0,  1,  2,  3],
       [ 8,  9, 10, 11],
       [12, 13, 14, 15]])

如果不想对原数组进行修改，可以使用copy函数：

arr_new=arr.copy(arr)

再使用arr_new进行操作

2. 交换两列

arr_new=arr[:, [1,0,2,3]]
print(arr_new)
# 结果如下
>>>array([[ 1,  0,  2,  3],
       [ 5,  4,  6,  7],
       [ 9,  8, 10, 11],
       [13, 12, 14, 15]])