使用另一个列表中的值索引列表（python）[重复]

Question

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

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


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

我想要的输出是：

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

这是我'want'的元素列表

如何做到这一点？

Answer 1

您可以使用zip 逐个元素地遍历两个列表，然后如果idx 中的对应元素等于'want'，则保留mylist 中的元素

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