源数组:
In [215]: a
Out[215]:
array([[4, 8, 2, 0, 5],
[3, 1, 6, 8, 1],
[2, 2, 6, 0, 3],
[9, 7, 6, 7, 8],
[5, 8, 1, 1, 4]], dtype=int64)
使用 Numpy 索引:
In [218]: a[:, a[0].argsort()]
Out[218]:
array([[0, 2, 4, 5, 8],
[8, 6, 3, 1, 1],
[0, 6, 2, 3, 2],
[7, 6, 9, 8, 7],
[1, 1, 5, 4, 8]], dtype=int64)
使用熊猫:
In [212]: pd.DataFrame(a).sort_values(0, axis=1).values
Out[212]:
array([[0, 2, 4, 5, 8],
[8, 6, 3, 1, 1],
[0, 6, 2, 3, 2],
[7, 6, 9, 8, 7],
[1, 1, 5, 4, 8]], dtype=int64)