循环筛选DataFrame列未生效，问题出在哪里？

你遇到的问题核心是循环中的变量赋值不会修改原列表里的DataFrame对象。当你在循环里写df = df[["Top 2 Box", "Label"]]时，只是把循环变量df指向了一个新的DataFrame切片，原列表df_list里的df1、df2并没有被修改，它们还是原来的对象。

下面给出两种可行的解决方法：

方案1：原地修改原DataFrame #

直接在原DataFrame对象上修改列，无需重新赋值，会直接改变原对象：

方法A：用drop删除冗余列 #

df_list = [df1, df2]
for df in df_list:
    # 删除指定列之外的所有列，inplace=True表示原地修改
    df.drop(df.columns.difference(["Top 2 Box", "Label"]), axis=1, inplace=True)

# 此时原df1、df2已更新
print(df1.head())
print(df2.head())

方法B：直接替换列（更简洁） #

df_list = [df1, df2]
for df in df_list:
    # 直接替换原DataFrame的列，保持指定顺序
    df[:] = df[["Top 2 Box", "Label"]]

方案2：生成新的DataFrame列表 #

如果需要保留原DataFrame的备份，推荐创建新的筛选后列表：

df_list = [df1, df2]
# 用列表推导式生成筛选后的新列表
filtered_dfs = [df[["Top 2 Box", "Label"]] for df in df_list]

# 若要更新原变量，直接赋值即可
df1, df2 = filtered_dfs

print(df1.head())
print(df2.head())

方案选择建议 #

• 不需要保留原数据时，方案1的原地修改更节省内存；
• 需要保留原数据备份时，方案2更安全，避免误改原始数据。

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