Sorting arrays in NumPy by column

后端 未结 13 2572
既然无缘
既然无缘 2020-11-22 03:47

How can I sort an array in NumPy by the nth column?

For example,

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

13条回答
  •  予麋鹿
    予麋鹿 (楼主)
    2020-11-22 04:34

    Simply using sort, use coloumn number based on which you want to sort.

    a = np.array([1,1], [1,-1], [-1,1], [-1,-1]])
    print (a)
    a=a.tolist() 
    a = np.array(sorted(a, key=lambda a_entry: a_entry[0]))
    print (a)
    

提交回复
热议问题