如何拆分字典键的下划线分隔符并通过pandas生成二维表格

实现思路 #

拆分字典键后需要和对应的值绑定，整理为结构化列表后即可传入pandas处理，再通过透视操作生成你需要的二维表结构。

完整代码示例 #

import pandas as pd

# 原始字典
raw_dict = {"a_1":20,"b_1":5,"a_2":8,"b_2":7}

# 预处理键值对，拆分键并绑定对应值
processed_list = []
for key, value in raw_dict.items():
    # 按下划线拆分键为两部分
    key_part1, key_part2 = key.split("_")
    processed_list.append([key_part1, key_part2, value])

# 转换为DataFrame
df = pd.DataFrame(processed_list, columns=["row_key", "col_key", "val"])

# 透视生成二维表，row_key为行，col_key为列
result = df.pivot(index="row_key", columns="col_key", values="val")
# 若需要把行索引转为普通列，添加.reset_index()即可：
# result = df.pivot(index="row_key", columns="col_key", values="val").reset_index()

输出效果 #

最终生成的二维表结构如下：

如果需要调整行、列的命名规则，直接修改pivot方法对应的index、columns参数即可

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