Categories
不学无术

Python numpy 按矩阵的某一列排序

可以参考我在StackOverflow上的回答
sorting arrays in numpy by column
利用numpy的argsort对需要排序的列标记序号(索引),然后直接改变原始矩阵的索引就行啦~

a = np.array([[1,2,3],[4,5,6],[0,0,1]])
a_arg = np.argsort(a[:,1]) #按第'1'列排序
a = a[a_arg]