Question

假设我有一个index= np.array([4,2,3,1])的索引列表。我有一个二维数组blah，并希望使用blah提取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]]

将列表传递给数组

2 个答案: