Dada una lista y una máscara de bits, ¿cómo devuelvo los valores en los índices que son verdaderos?

Question

que empezar con la lista s siguiente y máscara de bits 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 

¿Cómo escribo alguna función apply_bitmask(s, b) para que vuelva

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

Answer 1

Python 3.1 itertools.compress (o Python 2.7's si no se ha actualizado todavía) hace exactamente eso (la lista es una comprensión muy cerca segundo):

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

Tenga en cuenta que esto produce un iterador, no una lista. Guarda la memoria, pero si necesita repetirla varias veces o usar índices, siempre puede usar list(itertools.compress(s, b)). Aún más corto.

Answer 2

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

Answer 3

Puede utilizar 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. 

Primero toma las dos listas originales, y zips, para que pueda obtener una lista (temporal - ¡todavía está dentro de la lista!) De pares de palabras y sus máscaras - algo así como [('baa',1), ('baa',0),...]. Entonces solo las palabras que tienen una máscara de 1 (if mask == 1) se agregan al newList.

Answer 4

otra toma en la lista de la comprensión, witout usando postal

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