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]])
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)