我想用另一个这样的列表来索引一个列表
L = ['a', 'b', 'c', 'd', 'e', 'f', 'g', 'h']
Idx = [0, 3, 7]
T = L[ Idx ]
T应该是一个包含'a','d','h‘的列表。
还有比这更好的方法吗?
T = []
for i in Idx:
T.append(L[i])
print T
# Gives result ['a', 'd', 'h']
发布于 2009-06-18 11:38:47
T = [L[i] for i in Idx]
发布于 2009-06-18 12:54:26
如果您使用numpy,则可以像这样执行扩展切片:
>>> import numpy
>>> a=numpy.array(['a', 'b', 'c', 'd', 'e', 'f', 'g', 'h'])
>>> Idx = [0, 3, 7]
>>> a[Idx]
array(['a', 'd', 'h'],
dtype='|S1')
...and可能要快得多(如果性能问题足以困扰于numpy导入)
发布于 2015-07-07 07:12:25
函数式方法:
a = [1,"A", 34, -123, "Hello", 12]
b = [0, 2, 5]
from operator import itemgetter
print(list(itemgetter(*b)(a)))
[1, 34, 12]
https://stackoverflow.com/questions/1012185
复制相似问题