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]])
From the Python documentation wiki, I think you can do:
a = ([[1, 2, 3], [4, 5, 6], [0, 0, 1]]); a = sorted(a, key=lambda a_entry: a_entry[1]) print a
The output is:
[[[0, 0, 1], [1, 2, 3], [4, 5, 6]]]