एक सूची और बिटमैस्क को देखते हुए, मैं उन सूचकांकों पर मूल्य कैसे वापस कर सकता हूं जो सच हैं?

Question

मैं निम्न सूची s के साथ शुरू और b bitmask:

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

अजगर 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 उन्हें एक साथ ले जाता है, ताकि आपको शब्दों के जोड़े और उनके मुखौटे की सूची (अस्थायी - यह अभी भी सूची कंप के अंदर है!) - [('baa',1), ('baa',0),...] जैसे कुछ। फिर केवल वे शब्द जिनके पास 1 (if mask == 1) का मुखौटा है, newList में जोड़े गए हैं।

Answer 4

एक और सूची समझ पर ले, ज़िप

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