Python如何正确移除拆分后字符串内的DataFrame列名

你的实现逻辑存在两处核心偏差，导致输出和预期不符：

• 循环嵌套的结果收集逻辑错误：你在每次遍历单个列名做替换时，都会把单次替换的中间结果直接追加到结果列表，既没有把同一子串上的多轮替换效果叠加，也没有去重收集最终处理结果，最终列表长度是「列名数量 × 分割后子串数量」，和实际输出的4个元素完全对应。
• 执行流错位：正确的顺序应该是「先取出一个分割后的子串 → 在这个子串上遍历所有列名完成全部替换 → 把处理完成的最终值存入结果列表」，而你现在的顺序是「取出一个列名 → 遍历所有子串做单次替换 → 把结果存入列表」，完全颠倒了两层循环的职责。

基础版本（适配当前场景，逻辑清晰易理解） #

核心是调整两层循环的顺序，先固定子串，再对其应用所有列名的替换规则，最后收集结果：

values = "dffa,c20"
columns = ['a','c']
# 按逗号分割得到独立子串列表
split_parts = values.split(',')
result = []
for part in split_parts:
    processed_str = part
    # 对当前子串，依次替换所有列名
    for col in columns:
        processed_str = processed_str.replace(col, '')
    # 全量替换完成后，再存入最终结果
    result.append(processed_str)
print(result)

运行后输出即为预期的 ['dff', '20']。

高效优化版本（适配列名较多的场景） #

当列名数量较多时，多次循环调用replace会产生多次字符串遍历的开销，可以用正则表达式一次性匹配所有列名完成替换，减少循环开销：

import re
values = "dffa,c20"
columns = ['a','c']
# 转义列名中的正则特殊字符，避免语法错误，构造或关系的匹配规则
remove_pattern = re.compile('|'.join(map(re.escape, columns)))
# 分割子串后一次性完成全列名替换
result = [remove_pattern.sub('', part) for part in values.split(',')]
print(result)

注意：如果列名存在包含关系（比如列名同时存在a和aa），建议把列名按长度从长到短排序后再构造正则匹配规则，避免先替换短列名时破坏长列名的匹配逻辑。

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