Python-按列对NumPy中的数组排序

发布于 2021-02-02 23:18:32

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

例如,

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

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

array([[7, 0, 5],
       [9, 2, 3],
       [4, 5, 6]])
关注者
0
被浏览
113
1 个回答
  • 面试哥
    面试哥 2021-02-02
    为面试而生,有面试问题,就找面试哥。

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

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

    如果你最初没有使用字段定义数组,那么“正确”的方法将非常丑陋。

    作为一个简单的示例,对其进行排序并返回副本:

    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']来对第二列,第三列,第一列进行排序。



  • 面试哥
    面试哥 2021-02-02
    为面试而生,有面试问题,就找面试哥。

    我想这可行: a[a[:,1].argsort()]

    这表示的第二列,a并据此对其进行排序。



知识点
面圈网VIP题库

面圈网VIP题库全新上线,海量真题题库资源。 90大类考试,超10万份考试真题开放下载啦

去下载看看