gpt4 book ai didi

python - 像表格一样对 numpy 数组进行排序

转载 作者:太空狗 更新时间:2023-10-29 21:21:25 26 4
gpt4 key购买 nike

我有一个 list

[[0, 3], [5, 1], [2, 1], [4, 5]]

我使用 numpy.array 将其制作成一个数组:

[[0 3]
[5 1]
[2 1]
[4 5]]

如何像表格一样对它进行排序?特别是,我想按升序对第二列进行排序,然后通过按升序对第一列进行排序来解决任何关系。因此我希望:

[[2 1]
[5 1]
[0 3]
[4 5]]

如有任何帮助,我们将不胜感激!

最佳答案

参见 http://docs.scipy.org/doc/numpy/reference/generated/numpy.lexsort.html#numpy.lexsort

特别是在你的情况下,

import numpy as np
x = np.array([[0,3],[5,1],[2,1],[4,5]])
x[np.lexsort((x[:,0],x[:,1]))]

输出

array([[2,1],[5,1],[0,3],[4,5]])

关于python - 像表格一样对 numpy 数组进行排序,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/8153540/

26 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com