Question

我虽然这很容易，但我有点挣扎。我的数据结构如下

array([[  5.        ,   3.40166205],
   [ 10.        ,   2.72778882],
   [ 15.        ,   2.31881804],
   [ 20.        ,   2.50643777],
   [  1.        ,   3.94076063],
   [  2.        ,   3.80598599],
   [  3.        ,   3.67121134],
   [  6.        ,   3.2668874 ],
   [  7.        ,   3.13211276],
   [  8.        ,   2.99733811],
   [  9.        ,   2.86256347],
   [ 11.        ,   2.64599467],
   [ 12.        ,   2.56420051],
   [ 13.        ,   2.48240635],
   [ 14.        ,   2.4006122 ],
   [ 16.        ,   1.8280531 ],
   [ 17.        ,   1.74625894],
   [ 18.        ,   1.66446479],
   [ 19.        ,   1.58267063],
   [ 20.        ,   1.50087647]])

我想在第一列上对它进行排序......所以它的排序方式如下：

array([[1.  ,  3.9], 
       [2.  ,  3.8], 
       ...  , 
       [20. ,  1.5]])

np.sort似乎没有效果，因为它将数组移动到平面结构。我还使用了itemgetter

from operator import itemgetter
sorted(data, key=itemgetter(1))

但这并没有给我我想要的输出。

帮助表示赞赏！

Answer 1

这是一个常见的numpy成语。您可以使用argsort（在第一列）+ numpy索引 -

x[x[:, 0].argsort()]

array([[  1.        ,   3.94076063],
       [  2.        ,   3.80598599],
       [  3.        ,   3.67121134],
       [  5.        ,   3.40166205],
       [  6.        ,   3.2668874 ],
       [  7.        ,   3.13211276],
       [  8.        ,   2.99733811],
       [  9.        ,   2.86256347],
       [ 10.        ,   2.72778882],
       [ 11.        ,   2.64599467],
       [ 12.        ,   2.56420051],
       [ 13.        ,   2.48240635],
       [ 14.        ,   2.4006122 ],
       [ 15.        ,   2.31881804],
       [ 16.        ,   1.8280531 ],
       [ 17.        ,   1.74625894],
       [ 18.        ,   1.66446479],
       [ 19.        ,   1.58267063],
       [ 20.        ,   2.50643777],
       [ 20.        ,   1.50087647]])

按其中一列对2D NumPy数组进行排序

1 个答案: