基于另一个"True"/"False"列表的列表的选择元素

rez*_*nik 2 python list

我有两个相同长度的列表.第一个包含字符串.第二个-字符串,可以是'True''False'.

如果第二个列表的第n个元素是'True',我想将第一个列表的第n个元素附加到另一个列表.

所以,如果我有:

列表1:

('sth1','sth2','sth3','sth4') 
Run Code Online (Sandbox Code Playgroud)

列表2:

('True','False','True','False')

结果应该是List3:

('sth1','sth3').

如何以这种方式交叉两个列表?

Reu*_*ani 15

使用zip:

result = [x for x, y in zip(xs, ys) if y == 'True']
Run Code Online (Sandbox Code Playgroud)

例:

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']
Run Code Online (Sandbox Code Playgroud)


小智 8

或使用numpy:

import numpy as np
filtered = np.array(List1)[np.array(List2)]
Run Code Online (Sandbox Code Playgroud)

顺便说一句,这仅在List2内的元素为True / False时才有效,如果它们为“ True” /“ False”则不起作用。