将字典列表按键分组转换为NumPy数组教程-Python教程-PHP中文网

将字典列表按键分组转换为NumPy数组教程

DDD

发布： 2025-11-13 10:41:10

原创

199人浏览过

将字典列表按键分组转换为NumPy数组教程

本教程详细介绍了如何将包含单个键值对的字典列表高效转换为一个以原始键为索引、值为对应numpy数组的字典。通过迭代和动态分组，该方法能够将散布的数据聚合为结构化的数值数组，便于后续的数据分析和处理。

在数据处理和分析中，我们经常会遇到需要将特定格式的数据转换为更适合数值计算和分析的结构。一个常见的场景是将包含多个单键字典的列表，按照字典的键进行分组，并将每个键对应的所有值集合成一个NumPy数组。这种转换对于后续的统计分析、机器学习特征工程等任务至关重要。

核心转换方法

实现这一转换的核心思路是遍历原始字典列表，动态地将相同键的值收集起来，最终将这些收集到的值列表转换为NumPy数组。

步骤分解：

初始化数据结构： 创建一个空的字典，用于存储按键分组后的值。这个字典的键将是原始字典的键，而值将是一个列表，用于临时存放所有对应的值。
遍历并分组： 遍历输入的字典列表。对于每个字典，它只包含一个键值对。提取这个键和值。
- 检查当前键是否已存在于我们创建的分组字典中。
- 如果存在，将当前值追加到该键对应的列表中。
- 如果不存在，则以当前键为新键，创建一个新的列表，并将当前值作为第一个元素放入其中。
转换为NumPy数组： 完成所有值的分组后，再次遍历分组字典。对于字典中的每个键值对，将其值（此时是一个Python列表）转换为NumPy数组。

示例代码

以下是实现上述转换的Python代码示例：

import numpy as np

# 原始数据：包含单个键值对的字典列表
data_list = [{'Cool': 128}, {'Cool': 51}, {'Hot': 75}, {'Hot': 62}, {'Archive': 144}, {'Archive': 12}]

# 步骤1 & 2: 初始化分组字典并进行数据分组
grouped_values = {}
for item_dict in data_list:
    # 假设每个字典只包含一个键值对
    for key, value in item_dict.items():
        if key in grouped_values:
            grouped_values[key].append(value)
        else:
            grouped_values[key] = [value]

# 步骤3: 将分组后的列表转换为NumPy数组
result_arrays = {}
for key, value_list in grouped_values.items():
    result_arrays[key] = np.array(value_list)

# 打印结果以验证
print("转换结果:")
for key, np_array in result_arrays.items():
    print(f'"{key}": {np_array}')

# 预期输出:
# "Cool": [128  51]
# "Hot": [75 62]
# "Archive": [144  12]

登录后复制

代码解析

import numpy as np: 导入NumPy库，用于创建和操作数值数组。
data_list: 定义了我们要处理的原始数据结构，一个包含多个单键字典的列表。
grouped_values = {}: 创建一个空字典grouped_values，它将作为中间容器，临时存储每个键对应的所有数值列表。
外层for item_dict in data_list:：逐一遍历data_list中的每一个字典。
内层for key, value in item_dict.items():：由于输入数据格式的特点，每个item_dict只包含一个键值对，这个循环用于方便地提取该键和值。
if key in grouped_values:：检查当前提取到的key是否已存在于grouped_values字典中。
- 如果存在，grouped_values[key].append(value)：将当前value追加到该key对应的现有列表中。
- 如果不存在，grouped_values[key] = [value]：以当前key为键，创建一个新的列表，并将当前value作为列表的第一个元素。
result_arrays = {}: 创建最终存储NumPy数组的字典。
for key, value_list in grouped_values.items():: 遍历完成分组后的grouped_values字典。
result_arrays[key] = np.array(value_list)：将每个键对应的Python列表（value_list）转换为NumPy数组，并将转换后的数组存储到result_arrays中，键保持不变。

注意事项与优化

数据结构假设： 本教程的方法是基于输入列表中的每个字典都只包含一个键值对的假设。如果字典可能包含多个键值对，则需要调整逻辑来决定如何处理多余的键值对或仅选择特定的键。

快转字幕
新一代 AI 字幕工作站，为创作者提供字幕制作、学习资源、会议记录、字幕制作等场景，一键为您的视频生成精准的字幕。

357

查看详情
效率： 对于大规模数据，Python字典的查找和列表的append操作都是高效的。整个过程的时间复杂度大致与输入数据的大小成线性关系，即O(N)，其中N是原始字典列表中的元素总数。

使用collections.defaultdict进行简化： Python标准库中的collections.defaultdict可以进一步简化分组逻辑，避免显式地检查键是否存在，使代码更加简洁：

from collections import defaultdict
import numpy as np

data_list = [{'Cool': 128}, {'Cool': 51}, {'Hot': 75}, {'Hot': 62}, {'Archive': 144}, {'Archive': 12}]

# 使用 defaultdict 自动创建列表
grouped_values_defaultdict = defaultdict(list)
for item_dict in data_list:
    for key, value in item_dict.items():
        grouped_values_defaultdict[key].append(value)

# 使用字典推导式将列表转换为NumPy数组
result_arrays_defaultdict = {key: np.array(value_list) for key, value_list in grouped_values_defaultdict.items()}

print("\n使用 defaultdict 转换结果:")
for key, np_array in result_arrays_defaultdict.items():
    print(f'"{key}": {np_array}')

登录后复制

defaultdict(list)在访问一个不存在的键时会自动创建一个空列表，从而省略了if key in ... else ...的判断语句，提高了代码的简洁性。

总结

通过上述方法，我们能够有效地将散列在字典列表中的数据，按照其键进行聚合，并转换为结构化的NumPy数组字典。这种转换不仅提高了数据的可读性，更重要的是为后续的数值计算和科学分析奠定了基础。无论是使用基本的字典操作还是更高级的defaultdict，核心思想都是迭代、分组和类型转换，以适应不同的数据处理需求。掌握这种数据转换技巧，对于任何从事Python数据科学和工程的人员都至关重要。

以上就是将字典列表按键分组转换为NumPy数组教程的详细内容，更多请关注php中文网其它相关文章！