2

在 Perl 中,我可以使用索引列表轻松选择多个数组元素,例如

my @array = 1..11;
my @indexes = (0,3,10);
print "@array[ @indexes ]"; # 1 4 11

在 Python 中执行此操作的规范方法是什么?

4

3 回答 3

7

使用operator.itemgetter

from operator import itemgetter
array = range(1, 12)
indices = itemgetter(0, 3, 10)
print indices(array)
# (1, 4, 11)

然后随心所欲地呈现该元组......,例如:

print ' '.join(map(str, indices(array)))
# 1 4 11
于 2013-06-15T17:15:44.217 回答
4
>>> array = range(1, 12)
>>> indexes = [0, 3, 10]
>>> [array[i] for i in indexes]
[1, 4, 11]
>>>
>>> list(map(array.__getitem__, indexes))
[1, 4, 11]
于 2013-06-15T17:19:52.820 回答
1

使用numpy

>>> import numpy as np
>>> indexes = (0,3,10)
>>> x = np.arange(1,12)
>>> x [np.array(indexes)]
array([ 1,  4, 11])
于 2013-06-15T17:35:25.643 回答