如何使用Python Pandas转换指定字典列表格式？

原始输入 #

original_list = [
    {'id': 1, 'code': 'a'},
    {'id': 2, 'code': 'b'},
    {'id': 3, 'code': 'c'}
]

目标输出 #

[
    {1: 'a'},
    {2: 'b'},
    {3: 'c'}
]

实现代码 #

import pandas as pd

original_list = [
    {'id': 1, 'code': 'a'},
    {'id': 2, 'code': 'b'},
    {'id': 3, 'code': 'c'}
]

# 转成DataFrame做结构化处理
df = pd.DataFrame(original_list)

# 把id设为索引，提取code列转字典，再拆分为单个小字典的列表
result = [{idx: code} for idx, code in df.set_index('id')['code'].to_dict().items()]

print(result)

步骤说明 #

• 将原始字典列表转为DataFrame，利用Pandas的结构化数据处理能力简化操作
• 通过set_index('id')将id列设置为数据索引，确保后续生成字典时id作为键
• 提取code列并转换为字典，此时得到{1: 'a', 2: 'b', 3: 'c'}形式的整体字典
• 用列表推导式将大字典的每个键值对拆分为独立的小字典，最终得到目标格式的列表

内容的提问来源于stack exchange，提问作者Adi