Python列表元素迭代清洗：提取字符串后缀失败求助

我一眼就看到问题出在哪了——你把结果列表new的初始化语句new = []放在了for循环的内部！这意味着每次循环迭代时，new都会被重置为空列表，之前处理好的元素全部被清空，最后自然只留下了最后一个处理后的字符串。

修复后的代码 #

把new = []移到循环外面，确保整个循环过程中都往同一个列表里添加元素：

names = ['pipeline-1__YearBuilt', 'pipeline-1__NumberofBuildings', 'pipeline-1__NumberofFloors', 'pipeline-1__PropertyGFATotal', 'onehotencoder__x0_Low-Rise Multifamily', 'onehotencoder__x0_Medical Office', 'onehotencoder__x0_Mixed Use Property', 'onehotencoder__x0_Office', 'onehotencoder__x0_Other', 'onehotencoder__x0_Refrigerated Warehouse', 'onehotencoder__x5_Yes' ]

new = []  # 初始化移到循环外部，只执行一次
for n in names:
    if n.startswith('pipeline-1'):
        cleaned_name = n.split('__', 1)[-1]
    else:
        cleaned_name = n.split('__x', 1)[-1]
    new.append(cleaned_name)

# 输出结果
print(new)

执行后会得到完整的清洗后列表：

['YearBuilt', 'NumberofBuildings', 'NumberofFloors', 'PropertyGFATotal', '0_Low-Rise Multifamily', '0_Medical Office', '0_Mixed Use Property', '0_Office', '0_Other', '0_Refrigerated Warehouse', '5_Yes']

更简洁的写法（列表推导式） #

如果追求代码简洁，可以用Python的列表推导式一行搞定，逻辑和上面的循环完全一致：

new = [n.split('__', 1)[-1] if n.startswith('pipeline-1') else n.split('__x', 1)[-1] for n in names]

内容的提问来源于stack exchange，提问作者Sv3n-Sk4