Numpy:使用不同的索引数组多次选择行

时间:2017-05-03 00:09:06

标签: python arrays numpy

假设我有以下数组。

l = np.asarray([1,3,5,7])

Out[552]: array([1, 3, 5, 7])

我可以使用索引数组np.asarray([[0,1],[1,2]])选择行两次:

l[np.asarray([[0,1],[1,2]])]
Out[553]: 
array([[1, 3],
       [3, 5]])

如果索引数组在每行上的长度不同,则不起作用:

l[np.asarray([[1,3],[1,2,3]])]

Traceback (most recent call last):

  File "<ipython-input-555-3ec2ab141cd4>", line 1, in <module>
    l[np.asarray([[1,3],[1,2,3]])]
IndexError: arrays used as indices must be of integer (or boolean) type

此示例的我想要的输出是:

array([[3, 7],
       [3, 5, 7]])

有人可以帮忙吗?

2 个答案:

答案 0 :(得分:1)

我认为这是我能得到的最接近的。

import numpy as np
l = np.asarray([1, 3, 5, 7])
idx = [[1,3],[1,2,3]]
output = np.array([np.array(l[i]) for i in idx])
print output

结果:

[array([3, 7]) array([3, 5, 7])]

答案 1 :(得分:0)

如果您单独构建列表,则可以获得所需的结果。

<强>代码:

l = np.asarray([1, 3, 5, 7])

# Build a sample array
print(np.array([[3, 7], [3, 5, 7]]))   

# do the lookups into the original array 
print(np.array([list(l[[1, 3]]), list(l[[1, 2, 3]])]))

<强>结果:

[[3, 7] [3, 5, 7]]
[[3, 7] [3, 5, 7]]