我有两个相同长度的列表.第一个包含字符串.第二个-字符串,可以是'True'
或'False'
.
如果第二个列表的第n个元素是'True'
,我想将第一个列表的第n个元素附加到另一个列表.
所以,如果我有:
列表1:
('sth1','sth2','sth3','sth4')
列表2:
('True','False','True','False')
结果应该是List3:
('sth1','sth3')
.
如何以这种方式交叉两个列表?
使用zip:
result = [x for x, y in zip(xs, ys) if y == 'True']
例:
xs = ('sth1','sth2','sth3','sth4') ys = ('True','False','True','False') result = [x for x, y in zip(xs, ys) if y == 'True'] result ['sth1', 'sth3']
或使用numpy:
import numpy as np filtered = np.array(List1)[np.array(List2)]
顺便说一句,这仅在List2内的元素为True / False时才有效,如果它们为“ True” /“ False”则不起作用。