import numpy as np
import cv2
import matplotlib.pyplot as plt
data_file = "all_data.txt"
dataarr = np.loadtxt(data_file, str)
sortrow = dataarr[:,180]
sortrow = sortrow.astype(float)
dataarr = dataarr[sortrow.argsort()]
将数据从txt中按照str读入之后,按照第181列进行排序。
sortrow将data的第181列所有数据取出,得到一个一维数组,用astype将其类型从str转换成float,使用argsort()返回排好序的行号,从而达到将矩阵按照列排序的目标。