可以使用pandas库中的apply方法来实现该功能。这里提供一个示例代码,假设我们要查找DataFrame中是否存在'apple',如果存在,则新创建一列并赋值1,否则赋值0。
import pandas as pd
# 创建一个DataFrame
df = pd.DataFrame({'A': ['apple', 'orange', 'banana'],
'B': ['orange', 'banana', 'apple'],
'C': ['banana', 'apple', 'orange']})
# 定义一个函数,用于查找DataFrame中是否存在'apple'
def find_apple(row):
if 'apple' in row.values:
return 1
else:
return 0
# 使用apply方法,遍历每一行,并将结果保存到新列中
df['has_apple'] = df.apply(find_apple, axis=1)
# 输出结果
print(df)
输出结果为:
A B C has_apple
0 apple orange banana 1
1 orange banana apple 1
2 banana apple orange 1
可以看到,新列“has_apple”成功创建,并且根据每一行是否包含'apple'赋值为1或0。
上一篇:遍历多个列并将先前行中的值(递增)替换为这些列中的空值。
下一篇:遍历多个枚举