Как наложить два списка?

Question

[Иван Мельников]

Есть два списка:

d1 = [1, 0, 1, 0]
d2 = ['qwe', 'asd', 'zxc', 'wer']

Нужно получить список путем "наложения" данных списков, то есть в третьем списке должны остаться только те элементы из списка d2, для которых в списке d1 на соответствующих позициях стоит 1:
d3 = ['qwe', 'zxc']
Как это сделать наиболее естественным способом?

Answer 1

[aRegius]

itertools.compress

d3 = list(compress(d2, d1))

Answer 2

[shurshur]

d1 = [1, 0, 1, 0]
d2 = ['qwe', 'asd', 'zxc', 'wer']

d3 = [x[1] for x in filter(lambda x:x[0] == 1, zip(d1, d2))]

print (d3)

Ещё короче:

d3 = [d2[i] for i in range(0,len(d1)) if d1[i] == 1]

Comments

[Drill]

А если так? ))

d1 = [1, 0, 1, 1]
d2 = ['qwe', 'asd', 'zxc']

[shurshur]

Drill, по условиям предполагается что в d1 и d2 одинаковое количество элементов.

[Drill]

shurshur, если предполагается, тогда да )

[shurshur]

Drill, в принципе это тривиально решается, а ещё лучше просто поставить проверку в начале, чтобы при неравных длинах кидать исключение, что было бы весьма логично и позволило бы отлавливать возможные косяки.

Answer 3

[Алан Гибизов]

d1 = [1, 0, 1, 0]
d2 = ['qwe', 'asd', 'zxc', 'wer']
print([d2[i] for i, _ in enumerate(d2) if d1[i]])

Comments

[Drill]

Попробуем так? ))

d1 = [1, 0, 1, 0]
d2 = ['qwe', 'asd', 'zxc', 'wer', 'bnm']

[Алан Гибизов]

Drill, зачем? Ну, в смысле, понятно, что будет - индекс еггог. Но ведь такая задача и не ставилась. Всё согласно ТУ.

Answer 4

[Drill]

d3 = [i2 for i1,i2 in zip(d1,d2) if i1]