下面是一个示例,演示了如何遍历Pandas DataFrame的列,并根据满足条件的情况创建新列:
import pandas as pd
# 创建一个示例DataFrame
data = {'Name': ['John', 'Emma', 'Michael', 'Jessica'],
'Age': [25, 30, 35, 40],
'Salary': [50000, 60000, 70000, 80000]}
df = pd.DataFrame(data)
# 创建一个空的新列
df['New Column'] = ''
# 遍历DataFrame的列
for index, row in df.iterrows():
# 根据满足条件的情况设置新列的值
if row['Age'] >= 30 and row['Salary'] >= 60000:
df.at[index, 'New Column'] = 'Yes'
else:
df.at[index, 'New Column'] = 'No'
# 打印更新后的DataFrame
print(df)
在这个示例中,我们首先创建一个示例DataFrame,其中包含Name、Age和Salary列。然后,我们使用df['New Column'] = ''
创建一个空的新列。接下来,我们使用iterrows()
方法遍历DataFrame的每一行,并使用if
语句检查每一行的Age和Salary列是否满足条件。如果满足条件,我们使用at
方法将新列的值设置为"Yes",否则设置为"No"。最后,我们打印更新后的DataFrame。