Pandas DataFrame转指定结构字典:以首列ID为键其余字段为值
实现方案
有两种常用实现方式,可根据你的使用场景选择:
方案1:基于你已生成的dic1直接转换
你已经通过to_dict(orient = 'index')得到了符合键值结构的字典,直接将其转换为目标DataFrame即可:
import pandas as pd # dic1为你已生成的字典 result = pd.DataFrame(dic1.items(), columns=['ID', 'list'])
方案2:直接从原DataFrame生成(效率更高)
不需要先转全量字典,直接操作原df生成目标结构:
import pandas as pd # 先排除ID列,将剩余列转为每行对应的字典,列名转小写匹配你需要的键格式 df['list'] = df.drop('ID', axis=1).rename(str.lower, axis=1).to_dict('records') # 筛选得到两列的最终结果 result = df[['ID', 'list']]
执行上述任意一种方案后,result就是你需要的两列结构输出。
内容的提问来源于stack exchange,提问作者aditya
相关产品推荐
相关产品推荐

