将 numpy 数组转换为另一个具有列索引的数组的 Pythonic 方法

那么，根据这个输入，我想做的是:

a=np.array([[5, 1, 10], [2, 3, 4]]) 

转换成另一个np数组:

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

执行此操作的 pythonic 方法是什么？

编辑:我使用的是 1 索引，但 0 索引也可以。添加了更多上下文以进行澄清。修复了初始化中的拼写错误。

列索引代表项目。每个单元格中的值代表该项目的分数。我想要做的是根据分数对每一行进行降序排序，并获取索引以进行进一步处理。

所以，如果我有上面解释的输出，排序后它看起来像:

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

然后我可以获取索引:

[[2, 0, 1],[2, 1, 0]] 

谢谢

请您参考如下方法:

首先你的a表达式丢失了[]

In [231]: a=np.array([[5, 1, 10], [2, 3, 4]])  # add extra [] 
In [232]: a 
Out[232]:  
array([[ 5,  1, 10], 
       [ 2,  3,  4]]) 

列表理解是生成显示列表的最简单方法

In [233]: [[(n,i+1) for i,n in enumerate(row)] for row in a] 
Out[233]: [[(5, 1), (1, 2), (10, 3)], [(2, 1), (3, 2), (4, 3)]] 

我也可以通过连接 np.arange(1,4) 来实现，但内部元素不是元组。我必须使用结构化数组来获得这种显示。

这是我们通过连接获得的 3d 数组:

In [234]: np.array(_) 
Out[234]:  
array([[[ 5,  1], 
        [ 1,  2], 
        [10,  3]], 
 
       [[ 2,  1], 
        [ 3,  2], 
        [ 4,  3]]]) 

具有相同 tolist() 输出的结构化数组:

In [244]: alist=[[(n,i+1) for i,n in enumerate(row)] for row in a] 
In [245]: a3=np.array(alist, dtype='i,i') 
In [246]: a3 
Out[246]:  
array([[( 5, 1), ( 1, 2), (10, 3)], 
       [( 2, 1), ( 3, 2), ( 4, 3)]],  
      dtype=[('f0', '<i4'), ('f1', '<i4')]) 

直接构建:

In [254]: a2=np.zeros((a.shape[0],a.shape[1],2),a.dtype) 
In [255]: a2[:,:,0]=a 
In [256]: a2[:,:,1]=np.arange(1,4) 
In [257]: a2 
Out[257]:  
array([[[ 5,  1], 
        [ 1,  2], 
        [10,  3]], 
 
       [[ 2,  1], 
        [ 3,  2], 
        [ 4,  3]]]) 

或者对于结构化案例:

In [258]: a2=np.zeros((a.shape[0],a.shape[1]),dtype='i,i') 
In [259]: a2['f0']=a 
In [260]: a2['f1']=np.arange(1,4) 

---

相同的构造，但具有编辑中描述的排序值:

In [281]: idx=np.argsort(-a,axis=1) 
In [282]: a1 = -np.sort(-a,1) 
In [283]: a2=np.zeros((a.shape[0],a.shape[1]),dtype='i,i') 
In [285]: a2['f0']=a1 
In [286]: a2['f1']=idx 
In [287]: a2 
Out[287]:  
array([[(10, 2), ( 5, 0), ( 1, 1)], 
       [( 4, 2), ( 3, 1), ( 2, 0)]],  
      dtype=[('f0', '<i4'), ('f1', '<i4')])