详解Python字典的运算

详解Python字典的运算

问题描述

怎样在数据字典中执行一些计算操作(比如求最值、排序等)?

解决方案

有如下字典:

stocks = {
    "ACME": 45.23,
    "AAPL": 612.78,
    "IBM": 205.55,
    "HPQ": 37.20,
    "FB": 10.75
}

为了对字典值进行计算操作,通常需要使用zip()函数先将字典的键和值反转过来。比如:

# 取出value值最小的键值对
min_price = min(zip(stocks.values(), stocks.keys()))
print(min_price)
"""
结果:
(10.75, "FB")
"""
# 按value值的大小排序
stock_sorted = sorted(zip(stocks.values(), stocks.keys()))
print(stock_sorted)
"""
结果:
[(10.75, "FB"), (37.2, "HPQ"), (45.23, "ACME"), (205.55, "IBM"), (612.78, "AAPL")]
"""

执行这些计算的时候,需要注意zip()函数创建的是一个只能访问一次的迭代器。执行如下的代码就会产生错误:

price_and_name = zip(stocks.values(), stocks.keys())
min_price = min(price_and_name)
print(min_price)
"""
结果:
(10.75, "FB")
"""
max_price = max(price_and_name)
print(max_price)
"""
结果:
ValueError: max() arg is an empty sequence
"""

讨论

如果你在一个字典上执行普通的数学运算,你会发现它们仅仅作用于键,而不是值。比如:

min(stocks)  # return "AAPL"
max(stocks)  # return "IBM"

或许你会尝试使用字典的values()方法:

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

这也不是你想要的,你还想要知道对应的键的信息。

你可以在 min() 和 max() 函数中提供 key 函数参数来获取最小值或最大值对应的键的信息。比如:

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

但是这只返回了键,想要得到value值,还要进行一次查询:

min_value = stocks[min(stocks, key=lambda k: stocks[k])] # return 10.75
max_value = stocks[max(stocks, key=lambda k: stocks[k])] # return 612.78

前面的zip()函数方案通过将字典”反转”为 (值,键) 元组序列来解决了上述问题。 当比较两个元组的时候,值会先进行比较,然后才是键(如果值相同的话,就进行键的大小比较)。 这样的话你就能通过一条简单的语句就能很轻松的实现在字典上的求最值和排序操作了。

总结

本篇文章就到这里了,希望能够给你带来帮助,也希望您能够多多关注云海天教程的更多内容!