以下是一个示例代码,用于遍历列,找到选择项,并创建新列:
import pandas as pd
# 创建示例数据
data = {'Name': ['Tom', 'Nick', 'John', 'Sam'],
'Age': [20, 25, 30, 35],
'Gender': ['Male', 'Male', 'Female', 'Male'],
'Selection': ['Yes', 'No', 'Yes', 'No']}
df = pd.DataFrame(data)
# 创建新列
df['New Column'] = ''
# 遍历列
for index, row in df.iterrows():
# 找到选择项
if row['Selection'] == 'Yes':
# 在新列中添加值
df.at[index, 'New Column'] = 'Selected'
else:
df.at[index, 'New Column'] = 'Not Selected'
# 打印结果
print(df)
输出结果:
Name Age Gender Selection New Column
0 Tom 20 Male Yes Selected
1 Nick 25 Male No Not Selected
2 John 30 Female Yes Selected
3 Sam 35 Male No Not Selected
以上代码使用pandas库创建一个DataFrame,并在DataFrame中添加新列。然后,使用iterrows()方法遍历DataFrame的每一行,找到选择项并根据条件在新列中添加相应的值。