求助：Python-Pandas中DataFrame的while循环逻辑实现问题

嘿，我来帮你搞定这个需求！针对你给出的DataFrame和循环逻辑，我们可以用pandas结合while循环轻松实现，下面是具体步骤和代码：

步骤1：准备示例数据 #

首先我们先创建你提供的DataFrame：

import pandas as pd

# 初始化你的DataFrame
df = pd.DataFrame({'A': [5, 5, 10], 'B': [10, 10, 5]})
print("初始数据：")
print(df)

步骤2：编写while循环逻辑 #

核心思路是持续检查是否存在满足A < B的行，只要有就执行更新操作，直到没有符合条件的行为止。代码如下：

# 循环处理，直到所有行都不满足A < B
while (df['A'] < df['B']).any():
    # 生成布尔掩码，标记需要更新的行
    update_mask = df['A'] < df['B']
    # 对满足条件的行执行A和B的更新
    df.loc[update_mask, 'A'] += df.loc[update_mask, 'B'] / 2
    df.loc[update_mask, 'B'] /= 2

print("\n处理后的数据：")
print(df)

代码解释 #

• (df['A'] < df['B']).any()：判断DataFrame中是否至少有一行满足A < B，这是循环继续的条件；当所有行都不满足时，循环终止。
• update_mask：布尔索引，精准定位需要更新的行，避免修改不需要处理的数据。
• df.loc[update_mask, 'A'] += df.loc[update_mask, 'B'] / 2：对符合条件的行，将A的值加上B的一半。
• df.loc[update_mask, 'B'] /= 2：将符合条件的行的B值减半。

运行结果 #

执行上述代码后，你会得到这样的结果：

A    B
0  10  5.0
1  10  5.0
2  10  5.0

第一、二行原本A=5 < B=10，经过一次更新后A=10、B=5，不再满足条件；第三行本来就A > B，所以没有变化。

如果遇到需要多次迭代的场景（比如某行更新后仍然满足A < B），这个循环也会自动重复处理，直到所有行都符合要求。

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