Python List Comprehension

作者：星が降る│2020-07-20 17:13:26│巴幣：2│人氣：262

寫python的時候一直用到這個語法糖，但是看很多人好像不常用就隨手寫一篇文章了

接觸過python的應該都看過類似這行的code

[i*2 for i in range(6)]

剛接觸python的人可能會看不懂這行在幹麻，其實這是一種生成list的方式

圖片引用自wiki

上圖可能就常常看到了，其實這行code跟這個數學式的涵義差不多

可以把它想像成一個函數

後面是輸入的部分，並且可以對輸入做篩選

前面是輸出的部分，也就是我們所取得的list，可以對變量做操作(可以看做函數轉換)

再用一次前面的code

[i*2 for i in range(6)]

range(6)是輸入的list，而i代表range(6)中的元素，i*2則是對每個range(6)中的元素進行變換並輸出，所以得到的list就會是下面這行

[0, 2, 4, 6, 8, 10]

上面的範例沒有加入篩選，如果加入篩選

>>> [i*2 for i in range(6) if i%2 == 0]
[0, 4, 8]

上面的code中後面的if的作用是篩選輸入的元素，if i%2 == 0表示只有偶數才會通過篩選，因此輸入會變成[0, 2, 4]，輸出變成[0, 4, 8]

函數變換的部分也可以加入if else，注意這邊的if跟加在後面的if不同，這邊的是輸出轉換，所以一定要有else，也就是每個輸入的元素都要有對應的轉換

>>> [i if i%2 == 0 else i*3 for i in range(6)]
[0, 3, 2, 9, 4, 15]

>>> ["even" if i%2 == 0 else "odd" for i in range(6)]
['even', 'odd', 'even', 'odd', 'even', 'odd']

注意到下面那一個用法，輸出的轉換並不需要包含i，下面那行代表把奇數轉換成odd，偶數轉換成even

多變數及巢狀迴圈也是可行的

>>> [(x, y) for x in range(4) for y in range(3)]
[(0, 0), (0, 1), (0, 2), (1, 0), (1, 1), (1, 2), (2, 0), (2, 1), (2, 2), (3, 0), (3, 1), (3, 2)]

一般迴圈表示法

>>> tmp = list()
>>> for x in range(4):
...     for y in range(3):
...             tmp.append((x, y))
...
>>> tmp
[(0, 0), (0, 1), (0, 2), (1, 0), (1, 1), (1, 2), (2, 0), (2, 1), (2, 2), (3, 0), (3, 1), (3, 2)]

注意愈外層的要寫在愈前面

輸出的部分也可以為dictionary等資料結構

>>> [{'number': i, 'even': 1 if i%2 == 0 else 0} for i in range(6)]
[{'number': 0, 'even': 1}, {'number': 1, 'even': 0}, {'number': 2, 'even': 1}, {'number': 3, 'even': 0}, {'number': 4, 'even': 1}, {'number': 5, 'even': 0}]

依照這些規則可以變化出許多進階用法，可以節省程式的行數，不用寫一堆for

例如有一個list長這樣

['01-98', '02-65', '03-86']

每個字串前面是學號，後面是分數