如何使用Python中的索引列表获取列表切片?

时间:2013-06-15 17:14:52

标签: python slice

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

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

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

3 个答案:

答案 0 :(得分: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

答案 1 :(得分: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]

答案 2 :(得分:1)

使用numpy

>>> import numpy as np
>>> indexes = (0,3,10)
>>> x = np.arange(1,12)
>>> x [np.array(indexes)]
array([ 1,  4, 11])