1.17 从字典中提取子集

问题描述

你想构造一个字典，它是另外一个字典的子集。

解决方案

最简单的方式是使用字典推导式。例如：

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中的另一个推导式——字典推导式，并借之解决了从字典中提取子集的问题。