python字典同类相加统计

原创

mob64ca12ef217e 2024-12-09 06:21:32 ©著作权

©著作权归作者所有：来自51CTO博客作者mob64ca12ef217e的原创作品，请联系作者获取转载授权，否则将追究法律责任

Python 字典同类相加统计

Python 字典是一个强大的数据结构，允许我们以键值对的形式存储和访问数据。在实际应用中，我们常常需要对字典中同一类别的数据进行統計和合并。比如，我们有一个销售数据字典，其中包含了不同产品的销售额，有时我们希望将同一类产品的销售额进行相加统计。本文将详细介绍如何实现这一功能，并提供相应的代码示例。

1. 什么是 Python 字典

Python 字典（dict）是一种可变的、无序的数据结构，它由一系列键值对组成。每个键都是唯一的，用于访问与之对应的值。字典的基本操作包括添加、删除和查找元素等。

# 创建一个字典
sales_data = {
    "apple": 100,
    "banana": 150,
    "orange": 200,
    "banana": 50  # 注意：同名键会被覆盖
}
print(sales_data)

2. 同类相加统计的需求

在某些情况下，字典中可能会存储相同类别的键，例如多种水果的销售额。为了更好地分析这些数据，我们需要将相同类别的数值进行相加。

例如，假设我们的销售数据如下：

sales_data = {
    "apple": 100,
    "banana": 150,
    "orange": 200,
    "banana": 50,
    "apple": 150,
}

在这里，apple 和 banana 这两个键的值分别出现了两次，我们希望得到每种水果的总销售额。

3. 实现思路

我们可以使用一个空字典来保存结果。遍历原始字典中的每一个项目，如果这个键已经存在于结果字典中，就将对应的值相加；如果不存在，就直接添加到结果字典中。

以下是实现这一思路的代码：

# 原始销售数据
sales_data = {
    "apple": 100,
    "banana": 150,
    "orange": 200,
    "banana": 50,
    "apple": 150,
}

# 结果字典
result = {}

# 统计同类的销售额
for key, value in sales_data.items():
    if key in result:
        result[key] += value
    else:
        result[key] = value

print(result)

运行以上代码后，我们可以得到每种水果的总销售额：

{'apple': 250, 'banana': 200, 'orange': 200}

4. 用 mermaid 流程图展示实现步骤

在了解了实现的思路后，我们可以用 mermaid 的流程图来清晰地展示整个处理流程：

flowchart TD
    A[开始] --> B{遍历销售数据}
    B -->|找到键| C{键已存在?}
    C -->|是| D[将值相加]
    C -->|否| E[直接添加到结果]
    D --> B
    E --> B
    B -->|遍历完成| F[输出结果]
    F --> G[结束]

5. 扩展：处理更复杂的数据结构

在某些情况下，我们可能会处理嵌套字典或更复杂的数据结构。对于这些情况，我们可以使用递归函数或额外的类别来处理。

例如，如果我们的数据结构是这样的：

sales_data = {
    "fruits": {
        "apple": 100,
        "banana": 150,
        "orange": 200,
    },
    "vegetables": {
        "carrot": 50,
        "potato": 75,
        "carrot": 25,
    }
}

我们需要统计每个类别下各个项目的总和，可以使用如下的代码：

# 复杂的销售数据
sales_data = {
    "fruits": {
        "apple": 100,
        "banana": 150,
        "orange": 200,
    },
    "vegetables": {
        "carrot": 50,
        "potato": 75,
        "carrot": 25,
    }
}

# 结果字典
result = {}

# 统计函数
def collect_sales(data):
    for key, value in data.items():
        if isinstance(value, dict):  # 如果值是字典
            collect_sales(value)
        else:  # 否则，累加
            if key in result:
                result[key] += value
            else:
                result[key] = value

# 调用函数进行统计
collect_sales(sales_data)

print(result)

上述代码将正确处理嵌套字典，并输出合并后的结果。