下面是一个示例代码,用于遍历数据框中的元素,并根据条件添加一行:
import pandas as pd
# 创建示例数据框
data = {'A': [1, 2, 3], 'B': [4, 5, 6]}
df = pd.DataFrame(data)
# 遍历数据框中的元素
for index, row in df.iterrows():
if row['A'] == 2: # 根据条件判断是否添加一行
new_row = pd.Series({'A': 7, 'B': 8}) # 创建新行数据
df = df.append(new_row, ignore_index=True) # 将新行添加到数据框中
# 打印修改后的数据框
print(df)
输出结果为:
A B
0 1 4
1 2 5
2 3 6
3 7 8
在上面的示例中,我们使用iterrows()
方法遍历数据框中的每一行。然后,我们使用条件判断row['A'] == 2
来确定是否添加一行。如果满足条件,我们创建一个新的行数据new_row
,并使用append()
方法将其添加到数据框中。最后,我们打印修改后的数据框。