你想構造一個字典,它是另外一個字典的子集。
問題描述
解決方案
最簡單的方式是使用字典推導式。例如:
prices = {
'ACME': 45.23,
'AAPL': 612.78,
'IBM': 205.55,
'HPQ': 37.20,
'FB': 10.75
}
p1 = {key: value for key, value in prices.items() if value > 200}
print(p1)
"""輸出結果:
{'AAPL': 612.78, 'IBM': 205.55}
"""
tech_names = {'AAPL', 'IBM', 'HPQ', 'MSFT'}
p2 = {key: value for key, value in prices.items() if key in tech_names}
print(p2)
"""輸出結果:
{'AAPL': 612.78, 'IBM': 205.55, 'HPQ': 37.2}
"""
讨論
大多數情況下字典推導能做的,通過建立一個元組序列然後把它傳給
dict()
函數也能實作。例如:
p1 = dict((key, value) for key, value in prices.items() if value > 200)
但是,字典推導式表意更清晰,實際上運作速度也更快。
另外,第2個例子程式也可以這樣寫:
tech_names = { 'AAPL', 'IBM', 'HPQ', 'MSFT' }
p2 = { key:prices[key] for key in prices.keys() & tech_names }
但是這種方案運作效率很低。
總結
本節介紹了python中的另一個推導式——字典推導式,并借之解決了從字典中提取子集的問題。