以下是一个示例代码,演示了如何遍历DataFrame的行,并对两列的值进行求和,直到其中一列满足条件:
import pandas as pd
# 创建一个示例DataFrame
data = {'col1': [1, 2, 3, 4, 5],
'col2': [6, 7, 8, 9, 10]}
df = pd.DataFrame(data)
# 初始化求和值
sum_col1 = 0
sum_col2 = 0
# 遍历DataFrame的行
for index, row in df.iterrows():
# 对两列的值进行求和
sum_col1 += row['col1']
sum_col2 += row['col2']
# 判断条件是否满足
if sum_col1 > 5 or sum_col2 > 10:
break
print("sum_col1:", sum_col1)
print("sum_col2:", sum_col2)
该示例代码首先创建了一个示例DataFrame,然后初始化了两列的求和值。接下来,使用iterrows()
方法遍历DataFrame的行,对两列的值进行求和。在每次迭代中,判断两列的求和值是否满足条件,如果满足条件则跳出循环。最后,打印出两列的求和值。