Python列表解析配合if else

用習慣列表解析以後會以爲超級酷,因此在嘗試使用列表解析,把循環什麼的寫在一行裏面。使用if的時候何時必需要有else,何時能夠沒有else一直沒搞明白,直到今天!待我緩緩道來:app

列表解析總共有兩種形式:code

1. [i for i in range(k) if condition]:此時if起條件判斷做用,知足條件的,將被返回成爲最終生成的列表的一員。it

2. [i if condition else exp for exp]:此時if...else被用來賦值,知足條件的i以及else被用來生成最終的列表。io

以上狀況對多個for仍然成立。循環

# 當if放在判斷條件裏面的時候不須要配合else
# 原始寫法
list_b = []
for item in list_a:
    if item > 5:
        list_b.append(item)
list_a = [0, 1, 2, 3, 4, 5, 6, 7, 8, 9]

# 列表解析
list_b = [item for item in list_a if item > 5]  
# [6, 7, 8, 9]# 當用於賦值狀況時,須要配合else
# 原始寫法
list_c = []
for item in list_a:
    if item > 5:
        list_c.append(item)
    else:
        list_c.append(1)

# 列表解析
list_c = [item if item > 5 else 1 for item in list_a]  
# [1, 1, 1, 1, 1, 1, 6, 7, 8, 9]
相關文章
相關標籤/搜索