क्या किसी सूची (या dict) समझ में फ़ंक्शन की वापसी का परीक्षण करने का कोई तरीका है? मैं लिखने से बचना चाहता हूं कि:पायथन सूची समझ: परीक्षण फ़ंक्शन रिटर्न
lst = []
for x in range(10):
bar = foo(x)
if bar:
lst.append(bar)
और इसके बजाय एक सूची समझ का उपयोग करें। जाहिर है, मैं लिखना नहीं चाहता:
[foo(x) for x in range(10) if foo(x)]
तो?
[foo(x) for x in range(10) if ??? ]
कैसे
के बारे मेंfilter(None, map(foo, range(10)))
आप मध्यवर्ती सूची रखने के लिए नहीं करना चाहते हैं, itertools.imap() साथ map() बदलें। और itertools.ifilter() के साथ, पूरी चीज जनरेटर में बदल दी जा सकती है।
itertools.ifilter(None, itertools.imap(foo, range(10)))
बस मूल्यों की गणना करने और जेनरेटर से फ़िल्टर की गई सूची बनाने के लिए जनरेटर बनाएं।
उदाहरण:
# create generator via generator expression
results = (foo(x) for x in xrange(10))
# build result list, not including falsy values
filtered_results = [i for i in results if i]
संदर्भ के लिए:
क्यों नहीं है आप पहली बार जिस तरह से उपयोग करना चाहते हैं? यह उचित तरीका है। –
@ रोहित जैन - 'foo' के आधार पर, यह एक महंगा ऑपरेशन हो सकता है जिसे आप दो बार नहीं करना चाहते हैं। – mgilson
@ मिगिलसन: और पहले उदाहरण में, वह केवल एक बार ऐसा करता है। –