Python中根据关联列表对数据进行高效重排与分组-Python教程-PHP中文网

Python中根据关联列表对数据进行高效重排与分组

碧海醫心

发布： 2025-10-21 13:51:06

原创

653人浏览过

Python中根据关联列表对数据进行高效重排与分组

本文详细介绍了如何在python中根据一个列表的值对另一个关联列表进行重排和分组。通过利用`zip`函数将两个列表的元素配对，并结合字典（`dict.setdefault`或`collections.defaultdict`）进行高效的数据分组，最终实现对结果的排序输出。此方法适用于需要同步处理和组织多组关联数据的场景，提升数据处理的灵活性和效率。

在Python编程中，我们经常会遇到一种场景：有两个或多个列表，它们之间存在一对一的关联关系。例如，一个列表包含分类键，另一个列表包含对应的值。此时，我们可能需要根据分类键对这些关联数据进行分组，并按照键的顺序输出分组后的结果。本教程将详细介绍如何利用Python的内置函数和数据结构高效地实现这一需求。

核心概念：zip函数的使用

实现关联列表重排与分组的第一步，是有效地将两个（或多个）列表的对应元素配对起来。Python的内置函数zip()正是为此而生。zip()函数接收可迭代对象作为参数，将对应索引位置的元素打包成一个个元组，然后返回一个可迭代的zip对象。

例如，如果我们有两个列表：

key_list = [404, 403, 405, 404, 405]
value_list = [4, 5, 6, 7, 9]

登录后复制

使用zip(key_list, value_list)，我们可以得到如下配对： (404, 4), (403, 5), (405, 6), (404, 7), (405, 9)

这些元组构成了我们后续分组操作的基础。

立即学习“Python免费学习笔记（深入）”；

使用字典进行数据分组

一旦我们将关联数据配对，下一步就是根据键（key_list中的元素）对值（value_list中的元素）进行分组。Python的字典（dict）是实现这一目标最合适的数据结构，因为它能够将键映射到值，并且键是唯一的。

我们可以创建两个字典，一个用于存储键本身的分组，另一个用于存储对应值的分组。

方法一：使用 dict.setdefault()

dict.setdefault(key, default_value)方法是一个非常实用的工具。如果字典中存在key，则返回其对应的值；如果不存在，则插入{key: default_value}，并返回default_value。这使得我们可以在向字典中添加元素之前，自动初始化一个空列表作为值。

key_list = [404, 403, 405, 404, 405]
value_list = [4, 5, 6, 7, 9]

# 创建两个空字典用于存储分组结果
grouped_keys = {}
grouped_values = {}

# 遍历zip后的配对数据
for key, value in zip(key_list, value_list):
    # 使用setdefault初始化列表并添加元素
    grouped_keys.setdefault(key, []).append(key)
    grouped_values.setdefault(key, []).append(value)

print(f"Grouped Keys (unordered): {grouped_keys}")
print(f"Grouped Values (unordered): {grouped_values}")

登录后复制

输出示例：

序列猴子开放平台

具有长序列、多模态、单模型、大数据等特点的超大规模语言模型

查看详情

Grouped Keys (unordered): {404: [404, 404], 403: [403], 405: [405, 405]}
Grouped Values (unordered): {404: [4, 7], 403: [5], 405: [6, 9]}

登录后复制

请注意，字典的迭代顺序在Python 3.7+中是插入顺序，但在旧版本中可能不是固定的。为了确保最终输出的有序性，我们需要进行额外的排序步骤。

方法二：使用 collections.defaultdict

collections模块中的defaultdict是dict的一个子类，它接受一个工厂函数作为参数。当访问一个不存在的键时，defaultdict会自动调用工厂函数来生成该键的默认值。对于需要初始化空列表的场景，defaultdict(list)尤为方便。

from collections import defaultdict

key_list = [404, 403, 405, 404, 405]
value_list = [4, 5, 6, 7, 9]

# 创建两个defaultdict，默认值类型为列表
grouped_keys_dd = defaultdict(list)
grouped_values_dd = defaultdict(list)

# 遍历zip后的配对数据
for key, value in zip(key_list, value_list):
    # 直接添加元素，defaultdict会自动处理键不存在的情况
    grouped_keys_dd[key].append(key)
    grouped_values_dd[key].append(value)

print(f"Grouped Keys (defaultdict, unordered): {grouped_keys_dd}")
print(f"Grouped Values (defaultdict, unordered): {grouped_values_dd}")

登录后复制

输出示例与setdefault方法类似。

两种方法的比较

dict.setdefault(): 适用于不希望引入collections模块，或者需要在初始化时执行更复杂逻辑的场景。每次添加元素时都需要调用setdefault。
collections.defaultdict(list): 代码更简洁，当需要频繁向不存在的键添加元素时，减少了重复的条件判断或方法调用。推荐在需要自动初始化默认值的场景中使用。

对分组结果进行排序

字典在默认情况下不保证键的顺序（尽管Python 3.7+保持插入顺序）。如果我们需要按照键的自然顺序（例如，数字从小到大）输出分组结果，我们需要显式地对键进行排序，然后根据排序后的键来提取对应的值。

# 假设我们已经通过上述任一方法得到了grouped_keys和grouped_values字典
# 以defaultdict为例
from collections import defaultdict

key_list = [404, 403, 405, 404, 405]
value_list = [4, 5, 6, 7, 9]

grouped_keys_dd = defaultdict(list)
grouped_values_dd = defaultdict(list)

for key, value in zip(key_list, value_list):
    grouped_keys_dd[key].append(key)
    grouped_values_dd[key].append(value)

# 获取排序后的键列表
sorted_keys = sorted(grouped_keys_dd.keys())

# 根据排序后的键，构建最终的有序列表
final_grouped_keys = [grouped_keys_dd[k] for k in sorted_keys]
final_grouped_values = [grouped_values_dd[k] for k in sorted_keys]

print(f"Rearranged list: {final_grouped_keys}")
print(f"Rearranged folders: {final_grouped_values}")

登录后复制

预期输出：

Rearranged list: [[403], [404, 404], [405, 405]]
Rearranged folders: [[5], [4, 7], [6, 9]]

登录后复制

完整示例代码

下面是结合defaultdict和排序的完整代码示例，展示了如何将两个列表根据其中一个列表的元素进行分组和重排：

from collections import defaultdict

# 原始数据
second_lines_different_folders = [404, 403, 405, 404, 405]
different_lines_folders = [4, 5, 6, 7, 9]

# 使用defaultdict进行分组
# out1 将存储 second_lines_different_folders 中元素自身的分组
# out2 将存储 different_lines_folders 中对应元素的分组
out1 = defaultdict(list)
out2 = defaultdict(list)

# 遍历两个列表的配对元素
for key_val, associated_val in zip(second_lines_different_folders, different_lines_folders):
    out1[key_val].append(key_val)       # 根据 key_val 分组 key_val
    out2[key_val].append(associated_val) # 根据 key_val 分组 associated_val

# 获取排序后的键列表
sorted_order = sorted(out1.keys())

# 根据排序后的键，构建最终的有序列表
final_out1 = [out1[k] for k in sorted_order]
final_out2 = [out2[k] for k in sorted_order]

print(f"Rearranged list: {final_out1}")
print(f"Rearranged folders: {final_out2}")

登录后复制

注意事项与最佳实践

数据关联性: 确保两个列表之间确实存在一对一的关联关系，zip函数才能正确地将它们配对。
键的唯一性: 字典的键是唯一的。当使用键进行分组时，相同键的所有值都会被聚合到同一个列表中。
性能: 对于大型数据集，zip和字典操作通常效率很高。字典的查找和插入操作平均时间复杂度为O(1)。排序操作的时间复杂度为O(N log N)，其中N是不同键的数量。
可读性: 选择defaultdict通常能使代码更简洁、更易读，尤其是在需要频繁向字典中添加列表元素时。
错误处理: 在实际应用中，如果列表长度不匹配，zip函数会以最短的列表为准截断。如果需要严格的长度检查，应在调用zip之前进行。

总结

本教程详细阐述了如何在Python中利用zip函数、字典（dict.setdefault或collections.defaultdict）以及排序功能，对两个关联列表进行高效的数据重排与分组。掌握这些技巧，能够帮助开发者在处理复杂数据结构时，更加灵活和高效地组织和呈现数据，从而提升代码的专业性和实用性。

以上就是Python中根据关联列表对数据进行高效重排与分组的详细内容，更多请关注php中文网其它相关文章！