使用包含索引列表的变量检索列表元素python

Question

我正在尝试使用由索引列表组成的变量 flist 来检索 features 列表中的元素。

flist = [1, 3, 7]

features = ['a', 'b', 'c', 'd', 'e', 'f', 'g', 'h', 'i', 'j']

所需的输出：

['b', 'd', 'h']

有没有办法直接使用 features 从 flist 中检索元素？例如，当我尝试 features[flist[i] for i in range(0,len(flist))] 时，我收到“无效语法”错误。

谢谢

Answer 1

l = ['a', 'b', 'c', 'd', 'e', 'f', 'g', 'h', 'i', 'j']

idxs = [1, 3, 7]

print ([l[i] for i in idxs])

输出：

['b', 'd', 'h']

只需使用简单的列表推导式将 l 中的元素列表呈现在 idxs 中包含的索引处。