如何将现有DataFrame重塑为符合要求的新DataFrame

实现方案 #

你需要完成的是邻接矩阵格式转边列表格式的DataFrame重塑操作，使用Pandas可以快速实现，代码如下：

import pandas as pd

# 读取源数据文件，将第一列设为行索引
df = pd.read_csv("RelationMatrix.csv", index_col=0)

# 核心重塑逻辑：宽表转长表
target_df = df.stack().reset_index()
# 按需求修改列名，可自行调整
target_df.columns = ["人员A", "人员B", "关系类型"]

# 可选：过滤掉自己和自己的关联记录
target_df = target_df[target_df["人员A"] != target_df["人员B"]]

# 可选：如果是无向关系，可去除双向重复的记录
# target_df = target_df[target_df["人员A"] < target_df["人员B"]]

执行以上代码后得到的target_df就是你需要的目标格式。

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