小编典典

按列对 NumPy 中的数组进行排序

all

如何按第 n 列对 NumPy 中的数组进行排序?

例如,

a = array([[9, 2, 3],
           [4, 5, 6],
           [7, 0, 5]])

我想按第二列对行进行排序,以便返回:

array([[7, 0, 5],
       [9, 2, 3],
       [4, 5, 6]])

阅读 130

收藏
2022-03-16

共1个答案

小编典典

对于“正确”的方式,请参见numpy.ndarray.sort的 order
关键字参数

但是,您需要将数组视为具有字段的数组(结构化数组)。

如果您最初没有使用字段定义数组,那么“正确”的方式会非常难看......

举个简单的例子,对它进行排序并返回一个副本:

In [1]: import numpy as np

In [2]: a = np.array([[1,2,3],[4,5,6],[0,0,1]])

In [3]: np.sort(a.view('i8,i8,i8'), order=['f1'], axis=0).view(np.int)
Out[3]: 
array([[0, 0, 1],
       [1, 2, 3],
       [4, 5, 6]])

就地排序:

In [6]: a.view('i8,i8,i8').sort(order=['f1'], axis=0) #<-- returns None

In [7]: a
Out[7]: 
array([[0, 0, 1],
       [1, 2, 3],
       [4, 5, 6]])

据我所知,@Steve 确实是最优雅的方式......

此方法的唯一优点是“order”参数是用于排序搜索的字段列表。例如,您可以通过提供 order=[‘f1’,’f2’,’f0’]
按第二列、第三列、第一列排序。

2022-03-16