1. 程式人生 > >Python 解析式

Python 解析式

列表解析式

列表推導式(又稱列表解析式)提供了一種簡明扼要的方法來建立列表。其結構是在一箇中括號裡包含一個表示式,然後是一個for語句,然後是0個或多個for或者if語句。那個表示式可以是任意的,意思是你可以在列表中放入任意型別的物件。返回結果將是一個新的列表,在這個以if和for語句為上下文的表示式執行完成之後產生。

單個迴圈

有條件子句的列表解析式

numbers = [1, 2, 3, 4, 5]
doubled_odds = []
for n in numbers:
    if n % 2 == 1:
        doubled_odds.append(n * 2)

上面這段內碼表可以改寫為一個列表解析式:

numbers = [1, 2, 3, 4, 5]
doubled_odds = [n * 2 for n in numbers if n % 2 == 1]

無條件子句的列表解析式

doubled_numbers = []
for n in numbers:
    doubled_numbers.append(n * 2)

上面這段內碼表可以改寫為一個列表解析式:

doubled_numbers = [n * 2 for n in numbers]

巢狀迴圈

flattened = []
for row in matrix
: for n in row: flattened.append(n)

上面這段內碼表可以改寫為一個列表解析式:

flattened = [n for row in matrix for n in row]

提示:同樣地,上述原則也適用集合解析式和字典解析式

集合解析式

下面的程式碼提取單詞序列中每個單詞的首字母,建立了一個集合(set):

first_letters = set()
for w in words:
    first_letters.add(w[0])

同樣的程式碼可以改寫為集合解析式:

first_letters = {w[0] for w in words}

字典解析式

下面的程式碼將原有字典的鍵和值互換,從而建立了一個新的字典:

flipped = {}
for key, value in original.items():
    flipped[value] = key

同樣的程式碼可以改寫為字典解析式:

flipped = {value: key for key, value in original.items()}