Sorting a python array/recarray by column

前端 未结 5 1548
旧巷少年郎
旧巷少年郎 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 11:52

    To sort on the second column use itemgetter

    >>> from operator import itemgetter
    >>> data = [[5,2], [4,1], [3,6]]
    >>> sorted(data)
    [[3, 6], [4, 1], [5, 2]]
    >>> sorted(data,key=itemgetter(1))
    [[4, 1], [5, 2], [3, 6]]
    >>> 
    

提交回复
热议问题