在 Python 中，如何索引一个列表和另一个列表？

我想用另一个类似的列表索引一个列表

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']
218530 次浏览
小开
最佳答案 
T = [L[i] for i in Idx]
小开 
T = map(lambda i: L[i], Idx)
小开

如果使用 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')

... 并且可能更快(如果性能足够令人担忧，以至于不必为麻烦的导入操心的话)

小开

我对这些方法中的任何一种都不满意，所以我想出了一个 Flexlist类，它允许通过整数、切片或索引列表进行灵活的索引:

class Flexlist(list):
def __getitem__(self, keys):
if isinstance(keys, (int, slice)): return list.__getitem__(self, keys)
return [self[k] for k in keys]

举个例子，你可以把它用作:

L = Flexlist(['a', 'b', 'c', 'd', 'e', 'f', 'g', 'h'])
Idx = [0, 3, 7]
T = L[ Idx ]


print(T)  # ['a', 'd', 'h']
小开 
L= {'a':'a','d':'d', 'h':'h'}
index= ['a','d','h']
for keys in index:
print(L[keys])

我会使用一个 Dict add所需的 keys到 index

小开

功能性方法:

a = [1,"A", 34, -123, "Hello", 12]
b = [0, 2, 5]


from operator import itemgetter


print(list(itemgetter(*b)(a)))
[1, 34, 12]
小开

你也可以像下面这样结合使用 __getitem__方法和 map:

L = ['a', 'b', 'c', 'd', 'e', 'f', 'g', 'h']
Idx = [0, 3, 7]
res = list(map(L.__getitem__, Idx))
print(res)
# ['a', 'd', 'h']
小开

我的问题: 查找列表的索引。

L = makelist() # Returns a list of different objects
La = np.array(L, dtype = object) # add dtype!
for c in chunks:
L_ = La[c] # Since La is array, this works.

提交答案