1.17从字典中提取子集
问题
你想通过构造一个字典,它是另外又给字典的子集。
解决方案
大多数情况下字典推导能做的,通过创建一个元组序列然后把它传给dict()函数也能实现。比如:
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)
print(p1)
# ->{'AAPL': 612.78, 'IBM': 205.55}
但是,字典推导方式表意更清晰,并且实际上也会运行的更快一些(在这个例子中,实际测试几乎比dict()函数方式快整整一倍)。
有时候完成同一件事会有很多种方式。比如,第二个例子程序也可以像这样重写:
p2 = {key: prices[key] for key in prices.keys() & tech_names}
print(p2) # ->{'IBM': 205.55, 'AAPL': 612.78, 'HPQ': 37.2}
但是,运行时间测试显示这种方式大概比第一种方式慢1.6倍。如果对程序运行性能要求比较高的话,需要花点时间做计时的测试。

浙公网安备 33010602011771号