3 回答

TA貢獻(xiàn)1809條經(jīng)驗(yàn) 獲得超8個(gè)贊
c = some_condition # True or False
l = [1, 2] + [x for x in [3] if c]
print(l)
輸出 >>>
[1, 2, 3] # when c = True
[1, 2] # when c = False
你可以隨心所欲地?cái)U(kuò)展它
l = [1, 2] + [x for x in [3] if c] + [x for x in [4] if not c]
輸出 >>>
[1, 2, 3] # when c = True
[1, 2, 4] # when c = False

TA貢獻(xiàn)1827條經(jīng)驗(yàn) 獲得超8個(gè)贊
最接近的是使用單獨(dú)定義的生成器函數(shù):
def maker(condition):
yield 1
yield 2
if condition:
yield 3
yield 4
print(list(maker(True)))
print(list(maker(False)))
輸出:
[1, 2, 3, 4]
[1, 2]
也就是說(shuō),Python 并不真正適用于此類操作,因此它們會(huì)很笨重。通常,基于謂詞過(guò)濾列表更為慣用,或者創(chuàng)建相同形狀的布爾numpy數(shù)組并使用掩碼。
添加回答
舉報(bào)