ich mit der folgenden Liste starten s und Bitmaske b:Wie erhalten Sie eine Liste und eine Bitmaske, wie gebe ich die Werte an den Indizes zurück, die True sind?
s = ['baa', 'baa', 'black', 'sheep', 'have', 'you', 'any', 'wool']
b = [1, 0, 0, 0, 1, 1, 1, 0] # or any iterable with boolean values
Wie kann ich apply_bitmask(s, b) so eine Funktion schreiben, dass es
['baa', 'have', 'you', 'any']
Python 3.1 itertools.compress (oder Python 2.7's, wenn Sie noch nicht aktualisiert haben) macht genau das (die Liste Verständnis eine echte enge Sekunde ist):
import itertools
filtered = itertools.compress(s, b)
Beachten Sie, dass dies einen Iterator erzeugt, keine Liste. Speichert Speicher, aber wenn Sie es mehrmals durchlaufen oder Indizes verwenden müssen, können Sie immer list(itertools.compress(s, b)) verwenden. Noch kürzer.
[ item for item, flag in zip(s, b) if flag == 1 ]
kehrt können Sie list comprehensions verwenden:
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.
Das erste nimmt die ursprünglichen zwei Listen, und sie zusammen, so dass Sie eine (temporäre - das ist immer noch in der Liste comp!) Liste von Wortpaaren und ihre Masken - so etwas wie [('baa',1), ('baa',0),...]. Dann werden nur die Wörter mit der Maske 1 (if mask == 1) zu newList hinzugefügt.
Ein weiteres übernehmen Liste Verständnis, witout zip
newList = [item for i, item in enumerate(s) if b[i]]
Wow, das ist beeindruckend. Cheers für diesen Delnan - wusste nie über Itertools. – Stephen
@Stephen: Itertools ist in der Tat beeindruckend - nicht nur es setzt Iteratoren auf Steroiden (wie Dive Into Python 3 formuliert), viele von ihnen können in weniger als 5 Zeilen definiert werden. Fast fühlt sich an wie Haskell: D – delnan
Kann ich das mit '2.6.5' benutzen? – Kit