numpy快速交换行列

import numpy as np

“”"
快速加换两行两列
“”"

#交换两列第一列跟第三列
v1=np.arange(100)
v2=v1.reshape((20,5))
v3=v2[:,[0,3,2,1,4]]
print(v3)

“”"
快速交换两列
第一列跟第三列
“”"
print()
mask = list(range(5))#生成一个0到4的列表
#快速交换1和3元素的位置
mask[1],mask[3]=mask[3],mask[1]

v4=v2[:,mask]
print(v4)

“”"
快速交换两行第一行跟第3行
“”"

print()

mak=list(range(20))

temp = mak[1]
mak[1],mak[3]=mak[3],temp

print(v2[mak,:])
#print(v5)