如何通过步骤名称移除sklearn Pipeline中的指定步骤？

当Pipeline规模较大时，靠位置索引pipeline.steps.pop(n)移除步骤确实不方便，你可以通过以下两种方式直接按步骤名称移除：

方法一：找到名称对应的索引后移除 #

遍历Pipeline的steps列表，匹配目标步骤名称并获取其索引，再执行移除操作：

target_step_name = "your_step_name"
# 查找目标步骤的索引
step_index = next(i for i, (name, _) in enumerate(pipeline.steps) if name == target_step_name)
# 移除该步骤
pipeline.steps.pop(step_index)

方法二：重新构建steps列表（更直观） #

直接过滤掉目标名称的步骤，重新赋值给pipeline.steps：

target_step_name = "your_step_name"
# 保留所有名称不等于目标名称的步骤
pipeline.steps = [step for step in pipeline.steps if step[0] != target_step_name]

注意事项 #

• 操作后Pipeline的steps属性会直接修改，后续使用时要确保移除步骤不会破坏依赖关系（比如后续步骤需要用到被移除步骤的输出）。
• 如果目标名称不存在，第一种方法会抛出StopIteration异常，你可以按需添加异常处理逻辑：

try:
    step_index = next(i for i, (name, _) in enumerate(pipeline.steps) if name == target_step_name)
    pipeline.steps.pop(step_index)
except StopIteration:
    print(f"步骤 {target_step_name} 不存在于Pipeline中")

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