使用For循環查找列表中的第N個元素

Question

所以我想在另一個列表中找到第2，第3和第4個元素。

position = [2,3,4]

Sample_List = ['a','b','c','d','e']

循環將返回結果：

['c','d','e']

Answer 1

簡單：

the_elements = [Sample_List[i] for i in position]

Answer 2

你也可以使用

elements = map(lambda k:Sample_List[k], position)

在python3上，您需要將其轉換為列表

elements = list(map(lambda k:Sample_List[k], position))

Answer 3

另一種使用operator.itemgetter方法：

>>> from operator import itemgetter

>>> position = [2,3,4]
>>> Sample_List = ['a','b','c','d','e']
>>> itemgetter(*position)(Sample_List)
('c','d','e')