numpy：在单独的列中获取索引值

Question

如何尽可能有效地在单独的列中获取索引值？我知道如何在循环中执行此操作，但我想知道还有其他方法吗？

来自这个ndarray

[[ 0.71587892  0.72278279 ]
 [ 0.72225173  0.73340414 ]
 [ 0.7259692   0.72862454 ]]

到这个

[[0   0.71587892  0.72278279 ]
 [1   0.72225173  0.73340414 ]
 [2   0.7259692   0.72862454 ]]

Answer 1

怎么样

np.column_stack((np.arange(len(a)), a))

其中a是您的初始数组？

看看下面的IPython会话：

In [1]: import numpy as np

In [2]: a = np.array([[0.71587892, 0.72278279],
   ...:  [ 0.72225173, 0.73340414],
   ...:  [ 0.7259692, 0.72862454]])

In [3]: np.column_stack((np.arange(len(a)), a))
Out[3]: 
array([[ 0.        ,  0.71587892,  0.72278279],
       [ 1.        ,  0.72225173,  0.73340414],
       [ 2.        ,  0.7259692 ,  0.72862454]])