python - 将列表传递给数组

Question

假设我有一个索引列表，它们是index= np.array([4,2,3,1]). 我有一个二维数组blah，想提取blahusing的行blah[index,:]。我知道你可以在 R 中做到这一点，但是你如何在 python/numpy 中做到这一点？

Answer 1

In [24]: index = np.array([4,2,3,1])

In [25]: blah = np.array([[1], [2], [3], [4], [5]])

In [26]: blah[index]
Out[26]:
array([[5],
       [3],
       [4],
       [2]])

Answer 2

如果您想使用二维数组“blah”执行此操作，您可以执行以下操作：

index= np.array([4,2,3,1])
blah=np.array([[1,2,3,4,5], [6,7,8,9,10]])
matrix=[]
for i in range(len(blah)):
    matrix.append(blah[i][index])

print np.array(matrix)

结果：

[[ 5  3  4  2]
 [10  8  9  7]]