如何在Python循环中为DataFrame新增填充变量名的列
给DataFrame添加标识列并合并的正确实现方法
我来帮你搞定这个问题!你遇到的核心问题是没法把DataFrame的变量名对应到新增列里——直接遍历DataFrame列表时,拿到的是DataFrame对象,不是变量名,而且之前的循环逻辑也有点混乱。下面给你两种实用的解决方法:
方法一:手动处理(适合少量DataFrame)
这种方式简单直接,适合只有两三个DataFrame的场景:
import pandas as pd import quandl # 获取原始数据 df = quandl.get("FRED/DEXBZUS") df2 = quandl.get("FRED/DEXBZUS") # 为每个DataFrame添加Variable列,填入对应的变量名 df['Variable'] = 'df' df2['Variable'] = 'df2' # 重置索引:原数据的日期是索引,需要转为普通列(对应你期望输出的Year列) df = df.reset_index() df2 = df2.reset_index() # 合并两个DataFrame combined_df = pd.concat([df, df2], ignore_index=True) # 查看结果 print(combined_df)
方法二:用字典批量处理(适合多个DataFrame)
如果之后要处理更多类似的DataFrame,用字典来管理变量名和数据源会更高效,避免重复代码:
import pandas as pd import quandl # 用字典存储变量名和对应的Quandl代码 data_config = { 'df': "FRED/DEXBZUS", 'df2': "FRED/DEXBZUS" # 可以继续添加更多,比如'df3': "FRED/XXX" } processed_dfs = [] for var_name, quandl_code in data_config.items(): # 获取数据 temp_df = quandl.get(quandl_code) # 添加标识列 temp_df['Variable'] = var_name # 重置索引 temp_df = temp_df.reset_index() # 存入列表等待合并 processed_dfs.append(temp_df) # 合并所有处理后的DataFrame combined_df = pd.concat(processed_dfs, ignore_index=True) print(combined_df)
为什么你的原代码没成功?
你的循环逻辑有两个关键问题:
- 遍历
[df, df2]时,i是DataFrame对象,不是变量名df或df2,所以没法直接赋值给variable列; - 嵌套循环
for i in df是在遍历DataFrame的列名(原df只有Value列),这完全偏离了你的需求,反而把数据结构搞乱了。
内容的提问来源于stack exchange,提问作者Cesar
相关产品推荐
相关产品推荐

