1.8.使用字典计算.md

使用字典计算

问题

你想在字典中执行大量计算（例如，最小值，最大值，排序，etc）

解决方法

假如有如下股票价格的字典：

prices = {
   'ACME': 45.23,
   'AAPL': 612.78,
   'IBM': 205.55,
   'HPQ': 37.20,
   'FB': 10.75
}

为了执行有用的计算，通常使用zip()方法转换字典中的key-value。以下代码能找出最高和最低价格的股票和其价格：

min_price = min(zip(prices.values(), prices.keys()))
# min_price is (10.75, 'FB')
max_price = max(zip(prices.values(), prices.keys()))
# max_price is (612.78, 'AAPL')

同样，可以使用zip()和sorted()对其排序：

prices_sorted = sorted(zip(prices.values(), prices.keys()))
# prices_sorted is [(10.75, 'FB'), (37.2, 'HPQ'),
#                   (45.23, 'ACME'), (205.55, 'IBM'),
#                   (612.78, 'AAPL')]

这么做的时候，请注意，zip()创建的迭代器只能使用一次，以下代码是错误的：

prices_and_names = zip(prices.values(), prices.keys())
print(min(prices_and_names))   # OK
print(max(prices_and_names))   # ValueError: max() arg is an empty sequence

讨论

如果对普通数据进行最小最大查询，你将发现它们只执行key，而不执行value：

min(prices)    # Returns 'AAPL'
max(prices)    # Returns 'IBM'

这不是你想要的，你可以这么做：

min(prices.values())  # Returns 10.75
max(prices.values())  # Returns 612.78

不幸的是，这也不是你想要的，因为你要知道是哪个key，添加key lambda函数可以解决：

min(prices, key=lambda k: prices[k])  # Returns 'FB'
max(prices, key=lambda k: prices[k])  # Returns 'AAPL'

然而，要找到最小值，还得执行额外的查找：

min_value = prices[min(prices, key=lambda k: prices[k])]

这个方法使用zip()函数通过把字典“转化”成序列来解决问题。对这些元组执行比较时，value会首先被比较，然后是key。它可以很容易的对字典内容进行排序和最值查询，这正是你想要的。

需要提醒的是这种key-value计算中，如果value值相同，那么key被用来决定返回的顺序：

>>> prices = { 'AAA' : 45.23, 'ZZZ': 45.23 }
>>> min(zip(prices.values(), prices.keys()))
(45.23, 'AAA')
>>> max(zip(prices.values(), prices.keys()))
(45.23, 'ZZZ')
>>>