Pandas动态选取多列触发KeyError问题求助

问题根源 #

你之前的错误在于把多列名列表拼接成了单个字符串，而Pandas的DataFrame索引需要的是列名组成的列表，不是拼接后的字符串。比如'Col1|Col2'会被当成一个独立的列名去查找，而你的DataFrame里根本没有这个列，自然触发KeyError。

正确解决方案 #

直接将动态列列表和固定列列表合并成一个新的列名列表，无需任何字符串拼接操作：

import pandas as pd

# 示例数据
df = {'Col1': ['first_value', 'second_value'],
      'Col2': ['first_value', 'second_value'] ,
      'Col3': ['first_value', 'second_value'] }
df = pd.DataFrame(df)

# 动态列列表
Dynamic = ['Col1','Col2']

# 合并动态列和固定列，生成完整的列名列表
selected_columns = Dynamic + ['Col3']

# 选取列
df = df[selected_columns]
print(df)

运行后会正确输出包含Col1、Col2、Col3的DataFrame。

为什么之前的单列场景可行？ #

当Dynamic是单元素列表（比如['Col2']）时，'|'.join(Dynamic)返回的是'Col2'，此时你构造的列表是['Col1','Col2','Col3']，正好是正确的列名列表，所以能正常运行。但多列时，拼接操作会把多个列名变成一个字符串，导致索引错误。

额外补充：更灵活的列选取方式 #

如果需要在固定列之间插入动态列（比如要['Col1'] + Dynamic + ['Col3']），同样直接用列表加法即可：

selected_columns = ['Col1'] + Dynamic + ['Col3']
df = df[selected_columns]

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