文章转载地址(转载一下也方便自己以后使用时查找)
https://www.cnblogs.com/liyuxia713/p/7082091.html
想要实现指定某一行或某一列排序,将其换到首行/末行,或者首列/末列后进行排序,再将其换回来即可
import numpy as np
>>> a
array([[ 2, 7, 4, 2],
[35, 9, 1, 5],
[22, 12, 3, 2]])
# 按最后一列顺序排序
>>> a[np.lexsort(a.T)]
array([[22, 12, 3, 2],
[ 2, 7, 4, 2],
[35, 9, 1, 5]])
# 按最后一列逆序排序
>>>a[np.lexsort(-a.T)]
array([[35, 9, 1, 5],
[ 2, 7, 4, 2],
[22, 12, 3, 2]])
# 按第一列顺序排序
>>> a[np.lexsort(a[:,::-1].T)]
array([[ 2, 7, 4, 2],
[22, 12, 3, 2],
[35, 9, 1, 5]])
# 按最后一行顺序排序
>>> a.T[np.lexsort(a)].T
array([[ 2, 4, 7, 2],
[ 5, 1, 9, 35],
[ 2, 3, 12, 22]])
按第一行顺序排序
>>> a.T[np.lexsort(a[::-1,:])].T
array([[ 2, 2, 4, 7],
[ 5, 35, 1, 9],
[ 2, 22, 3, 12]])