Python for循环删除DataFrame列满足条件时终止的实现方法

原代码存在的问题 #

• 运算符错误：Python中^是按位异或运算符，不是平方计算符号，平方运算需使用**2
• 列匹配规则错误：你的列名格式为alpha_0这类下划线连接格式，原正则写alpha 带空格无法匹配到目标列
• 缺少终止判断逻辑：没有在每次计算Beta后校验是否满足所有值<=1的终止条件，无法提前跳出循环

修改后可直接运行的代码 #

import pandas as pd

# 初始化示例DataFrame
df = pd.DataFrame({
    'alpha_0': [1, 2, 0],
    'alpha_1': [2, 0, 0],
    'alpha_2': [1, 3, 0],
    'alpha_3': [4, 8, 9]
})

n_alphas = 4
for alpha_idx in range(n_alphas):
    # 匹配当前序号对应的alpha列并删除
    target_cols = df.filter(regex=f'alpha_{alpha_idx}').columns
    df.drop(target_cols, axis=1, inplace=True)

    # 按公式正确计算每行Beta值
    beta = (df.sum(axis=1) ** 2) / 10
    print(f"删除alpha_{alpha_idx}列后各行列Beta值：\n{beta}\n")

    # 校验是否所有行Beta都<=1，满足条件立刻终止循环
    if (beta <= 1).all():
        print("校验通过：所有行Beta值均不超过1，终止循环")
        break

逻辑说明 #

• 每次删除1列后立刻重新计算所有行的Beta值
• 用(beta <=1).all()判断是否所有行都满足阈值要求，返回True时直接用break跳出for循环，不会继续执行后续删列操作
• 你提供的示例数据中，删除全部4列后每行求和值为0，Beta值为0，此时会触发终止条件。

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