来自两个列表的子集值

Question

我有两个清单：

a = ['E', 'E', 'E', 'E', 'E', 'E', 'E', 'C', 'C', 'C']

b = ['C', 'C', 'E', 'E', 'E', 'E', 'E', 'C', 'E', 'C']

我想对这些进行子集化以获得：

a = ['E', 'E', 'E', 'E', 'E', 'E', 'E']

b = ['C', 'C', 'E', 'E', 'E', 'E', 'E']

这样我保留第一个列表中的“E”，但使用第二个列表 b 中的索引获取相应的位置值。

我尝试使用：

a = [x for x in a if x == 'E']

b = [x for x in b if x == 'E']

但这并不能保持顺序/位置，而是从列表 b 中仅提取 E。

Answer 1

使用 zip 函数压缩然后像这样解压缩a = ['E', 'E', 'E', 'E', 'E', 'E', 'E', 'C', 'C', 'C']b = ['C', 'C', 'E', 'E', 'E', 'E', 'E', 'C', 'E', 'C']a, b = zip(*[(ai, bi) for ai, bi in zip(a,b) if ai=="E"])print(a)print(b)输出('E', 'E', 'E', 'E', 'E', 'E', 'E')('C', 'C', 'E', 'E', 'E', 'E', 'E')

Answer 2

简单的itertools.takewhile功能：from itertools import takewhilea = ['E', 'E', 'E', 'E', 'E', 'E', 'E', 'C', 'C', 'C']b = ['C', 'C', 'E', 'E', 'E', 'E', 'E', 'C', 'E', 'C']a_slice = list(takewhile(lambda x: x == 'E', a))b_slice = b[:len(a_slice)]print(a_slice)   # ['E', 'E', 'E', 'E', 'E', 'E', 'E']print(b_slice)   # ['C', 'C', 'E', 'E', 'E', 'E', 'E']

Answer 3

尝试这个a = [x for x in a if x == 'E']indexes = [i for i, x in enumerate(a) if x == "E"]b = [b[i] for i in indexes]