如何创建以列位置为键、DataFrame列名为值的字典
解决方法
要生成以列位置(从1开始计数)为键、列名列表为值的字典,可通过Python的pandas结合字典推导式快速实现。
步骤1:构造示例DataFrame
先还原你提供的目标DataFrame:
import pandas as pd data = { 'ID': ['p', 'q', 'r'], 'A': [1, 4, 4], 'B': [3, 3, 0], 'C': [2, 2, 9] } df = pd.DataFrame(data)
步骤2:生成目标字典
利用enumerate遍历列名,将索引+1转为字符串作为键,列名放入列表作为值:
result_dict = {str(idx + 1): [col] for idx, col in enumerate(df.columns)} print(result_dict)
输出结果
运行后会得到你需要的格式:
{'1': ['ID'], '2': ['A'], '3': ['B'], '4': ['C']}
逻辑说明
enumerate(df.columns)会同时返回列的索引(从0开始)和对应的列名idx + 1将索引转换为从1开始的列位置,转成字符串后作为字典的键[col]把单个列名包装成列表,匹配你要求的值格式
内容的提问来源于stack exchange,提问作者serlomu
相关产品推荐
相关产品推荐

