1. 程式人生 > >python三元表示式+列表生成式

python三元表示式+列表生成式

三元表示式長的樣子大概是這樣:

def max1(a,b):
    return a if a > b else b  #相當於C中的a>b?a:b

print(max1(4,5))  #5

然後列表生成式大概是這樣:

a = [1, 2, 3, 4, 5]
b = [i+1 for i in a if i > 3]
print(b)  #[5, 6]
然後三元表示式加入列表生成式裡面就很騷了。比如:
a = [1, 2, 3, 4, 5]
b = [i if i < 5 else 0 for i in a if i > 3]
print(b)  #[4, 0]
當然了,可以在列表生成式裡面加入各種各樣的騷氣函式,比如map函式,據說可以大大減少程式碼執行時間2333。emmmmm反正一行程式碼就能弄成很騷的列表。