其实很简单:
- 交换两行:
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进行操作
- 交换两列
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]])