Sorting arrays in NumPy by column

后端 未结 13 2535
既然无缘
既然无缘 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:36

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

提交回复
热议问题