Sorting a python array/recarray by column

前端 未结 5 1569
旧巷少年郎
旧巷少年郎 2020-12-02 11:33

I have a fairly simple question about how to sort an entire array/recarray by a given column. For example, given the array:

import numpy as np
data = np.a         


        
5条回答
  •  日久生厌
    2020-12-02 12:14

    Use data[np.argsort(data[:, 0])] where the 0 is the column index on which to sort:

    In [27]: import numpy as np
    
    In [28]: data = np.array([[5,2], [4,1], [3,6]])
    
    In [29]: col = 0
    
    In [30]: data=data[np.argsort(data[:,col])]
    Out[30]: 
    array([[3, 6],
           [4, 1],
           [5, 2]])
    

提交回复
热议问题