可以使用 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。

Pandas DataFrame:高效判断列值相等并添加新列

原文地址: https://www.cveoy.top/t/topic/nwFK 著作权归作者所有。请勿转载和采集!

免费AI点我,无需注册和登录