Pandas DataFrame:高效判断列值相等并添加新列
可以使用 Pandas 中的 apply 函数来实现这个功能,代码如下:
import pandas as pd
def check_equal(row):
if row['A'] == row['B']:
row['C'] = 1
return row
else:
return row
df = pd.DataFrame({'A': [1, 2, 3], 'B': [1, 3, 3]})
df['C'] = 0
df = df.apply(check_equal, axis=1)
这里定义了一个函数 check_equal,该函数接收一个 Series 对象作为输入,判断 'A' 和 'B' 列是否相等,如果相等则将 'C' 列赋值为 1,最后返回整个 Series 对象。然后使用 apply 函数将该函数应用到每一行上,最后得到更新后的 DataFrame。
原文地址: https://www.cveoy.top/t/topic/nwFK 著作权归作者所有。请勿转载和采集!