使用另一个列表中的值索引列表（python）

Question

假设我有一个字符串列表mylist，我想要从中提取满足另一个列表中条件的元素idx：

mylist = ['a','b','c','d']


idx = ['want','want','dont want','want']

我想要的输出是：

['a','b','d']

这是我'want'

的元素列表

怎么能这样做？

Answer 1

如果zip中的相应元素等于mylist idx按顺序跨越两个列表，然后保留'want'元素。 >

>>> [i for i, j in zip(mylist, idx) if j == 'want']
['a', 'b', 'd']