Учитывая список и битовую маску, как мне вернуть значения по индексам True?

Question

я начинаю со следующим списком s и Bitmask b:

s = ['baa', 'baa', 'black', 'sheep', 'have', 'you', 'any', 'wool'] 
b = [1, 0, 0, 0, 1, 1, 1, 0] # or any iterable with boolean values 

Как я пишу какую-то функцию apply_bitmask(s, b) так, что она возвращает

['baa', 'have', 'you', 'any'] 

Answer 1

Python 3.1 itertools.compress (или Python 2.7's, если вы еще не обновлен) делает именно это (список понимание является очень близко второй):

import itertools 
filtered = itertools.compress(s, b) 

Обратите внимание, что это производит итератор, а не список. Сохраняет память, но если вам нужно повторить ее несколько раз или использовать индексы, вы всегда можете использовать list(itertools.compress(s, b)). Еще короче.

Answer 2

[ item for item, flag in zip(s, b) if flag == 1 ] 

Answer 3

Вы можете использовать list comprehensions:

newList = [word for (word, mask) in zip(s,b) if mask] 
# Note: Could also use 'if mask == blah', if mask is not a boolean-compatible type. 

Сначала вы берете исходные два списка и zips их вместе, так что вы получаете список (временные - это все еще в списке comp!) Список пар слов и их масок - что-то вроде [('baa',1), ('baa',0),...]. Затем к newList добавляются только слова, содержащие маску 1 (if mask == 1).

Answer 4

Другой взять на список понимания, witout используя почтовый

newList = [item for i, item in enumerate(s) if b[i]]