以下是一个遍历数据框并创建新列的示例代码:
import pandas as pd
# 创建一个示例数据框
data = {'Name': ['John', 'Emma', 'Mike'],
'Age': [25, 30, 35],
'Salary': [50000, 60000, 70000]}
df = pd.DataFrame(data)
# 创建一个新列并对每行进行操作
df['Bonus'] = 0 # 创建一个名为'Bonus'的新列,并初始化为0
for index, row in df.iterrows():
if row['Age'] < 30:
df.at[index, 'Bonus'] = 1000 # 如果年龄小于30岁,则将'Bonus'列的值设为1000
else:
df.at[index, 'Bonus'] = 500 # 否则,将'Bonus'列的值设为500
# 打印结果
print(df)
运行上述代码将输出如下结果:
Name Age Salary Bonus
0 John 25 50000 1000
1 Emma 30 60000 500
2 Mike 35 70000 500
在这个示例中,我们创建了一个新列'Bonus',并对每行进行操作。如果年龄小于30岁,则将'Bonus'的值设为1000,否则设为500。
上一篇:遍历数据框
下一篇:遍历数据框并导出数据