Numpy按列按周期对矩阵进行排序

Question

我有这个 numpy 数组：

array(   
[[  1.        ,   9.        , 565.98653513],  
[  1.        ,   1.        , 973.18466261],  
[  1.        ,  25.        , 803.17747373],  
[  2.        ,   9.        ,  82.56336897],  
[  2.        ,   1.        , 104.69517373],  
[  2.        ,  25.        , 627.01127514],  
[  3.        ,  21.        , 334.07622382],  
[  3.        ,  34.        , 921.37623107],  
[  3.        ,  20.        , 342.08177942],

... ...  ... ...

[ 10.        ,   7.        , 424.29338026],  
[ 10.        ,   0.        , 232.71475407],  
[ 10.        ,   1.        , 330.44846202]])

但我想按循环中的第一列对矩阵进行排序：1、2、3、...、10。它应该是这样的：

array(   
[[  1.        ,   9.        , 565.98653513],  
[  2.        ,   9.        ,  82.56336897],  
[  3.        ,  21.        , 334.07622382],  

... ... ... ...  

[ 10.        ,   7.        , 424.29338026], 
[  1.        ,   1.        , 973.18466261],  
[  2.        ,   1.        , 104.69517373],  
[  3.        ,  34.        , 921.37623107],  

... ... ... ...  

[ 10.        ,   0.        , 232.71475407],
[  1.        ,  25.        , 803.17747373],  
[  2.        ,  25.        , 627.01127514],  
[  3.        ,  20.        , 342.08177942],  

... ... ... ...  

[ 10.        ,   1.        , 330.44846202]])  

我该怎么做？

我正在考虑将其转换为数据框（即 pandas）以获得更多排序选项，然后转换回数组...但我没有看到直接的函数来执行此操作。

感谢任何帮助或想法。

Answer 1

假设你的数组被命名为x:

y = [x[i::10] for i in range(int(len(x)/10))]
y = np.array(y)
y.reshape(x.shape)
print(y)

x[i:j:k] 表示法表示x 从 i 到 j，步骤为 k。所以x[i::10] 的意思是x 从 i 到 end 步骤 10。

查看更多here。