推導式(又稱解析式)是Python的一種獨有特性,是可以從一個數據序列構建另一個新的數據序列的結構體。
共有三種推導,在Python2和3中都支持:
- 列表(list)推導式
- 字典(dict)推導式
- 集合(set)推導式
1 列表推導式(list comprehensions)
列表推導式(又稱列表解析式)提供了一種簡明扼要的方法來創建列表。
它的結構是在一個中括號里包含一個表達式,然后是一個for語句,然后是0個或多個for或者if語句。這個表達式可以是任意的,你可以在列表中放入任意類型的對象。返回結果將是一個新的列表,在這個以if和for語句為上下文的表達式運行完成之后產生。
標準規范:
variable = [out_exp for out_exp in input_list if out_exp == 2]
這里是另外一個簡明例子:
multiples = [i for i in range(20) if i % 3 == 0]
print(multiples) # output: [0, 3, 6, 9, 12, 15, 18]
這將對快速生成列表非常有用。有些人甚至更喜歡使用它而不是filter函數。
列表推導式在有些情況下超贊,特別是當你需要使用for循環來生成一個新列表。舉個例子,你通常會這樣做:
squared = []
for x in range(10):
squared.append(x**2)
你可以使用列表推導式來簡化它,就像這樣:
squared = [x**2 for x in range(10)]
2 字典推導式(dict comprehensions)
字典推導和列表推導的使用方法是類似的。
mcase = {'a': 10, 'b': 34, 'A': 7, 'Z': 3}
mcase_frequency = {
k.lower(): mcase.get(k.lower(), 0) + mcase.get(k.upper(), 0)
for k in mcase.keys()
}
print(mcase_frequency) # output:{'a': 17, 'z': 3, 'b': 34}
在上面的例子中我們把同一個字母但不同大小寫的值合并起來了。就我個人來說沒有大量使用字典推導式。
你還可以快速對換一個字典的鍵和值:
{v: k for k, v in some_dict.items()}
3 集合推導式(set comprehensions)
它跟列表推導式也是類似的,唯一的區別在于它使用大括號{ }。
舉個例子:
squared = {x**2 for x in [1, 1, 2]}
print(squared) # output:set([1, 4])