1. 程式人生 > 其它 >python 各種推導式玩法

python 各種推導式玩法

技術標籤:python

推導式套路

除了最簡單的列表推導式和生成器表示式,其實還有字典推導式、集合推導式等等。

下面是一個以列表推導式為例的推導式詳細格式,同樣適用於其他推導式。

variable = [out_exp_res for out_exp in input_list if out_exp == 2]
  out_exp_res:  列表生成元素表示式,可以是有返回值的函式。
  for out_exp in input_list:  迭代input_list將out_exp傳入out_exp_res表示式中。
  if out_exp == 2:  根據條件過濾哪些值可以。

列表推導式

例一:30以內所有能被3整除的數

'''
遇到問題沒人解答?小編建立了一個Python學習交流QQ群:778463939
尋找有志同道合的小夥伴,互幫互助,群裡還有不錯的視訊學習教程和PDF電子書!
'''
multiples = [i for i in range(30) if i % 3 is 0]
print(multiples)
# Output: [0, 3, 6, 9, 12, 15, 18, 21, 24, 27]

例二:30以內所有能被3整除的數的平方

def squared(x):
    return x*x
multiples = [squared(i) for i in
range(30) if i % 3 is 0] print(multiples)

例三:找到巢狀列表中名字含有兩個‘e’的所有名字

names = [['Tom', 'Billy', 'Jefferson', 'Andrew', 'Wesley', 'Steven', 'Joe'],
         ['Alice', 'Jill', 'Ana', 'Wendy', 'Jennifer', 'Sherry', 'Eva']]

print([name for lst in names for name in lst if name.count('e') >= 2])  # 注意遍歷順序,這是實現的關鍵

字典推導式

例一:將一個字典的key和value對調

'''
遇到問題沒人解答?小編建立了一個Python學習交流QQ群:778463939
尋找有志同道合的小夥伴,互幫互助,群裡還有不錯的視訊學習教程和PDF電子書!
'''
mcase = {'a': 10, 'b': 34}
mcase_frequency = {mcase[k]: k for k in mcase}
print(mcase_frequency)

例二:合併大小寫對應的value值,將k統一成小寫

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)

集合推導式

例:計算列表中每個值的平方,自帶去重功能

squared = {x**2 for x in [1, -1, 2]}
print(squared)
# Output: set([1, 4])