使用Python的pandas库,可以使用.apply()和lambda函数来遍历两列,检查元素的相等性,并创建新列。
代码示例:
import pandas as pd
# 创建示例数据
df = pd.DataFrame({'A': [1, 2, 3, 4], 'B': [1, 3, 3, 5]})
# 定义lambda函数,返回True或False
check_identity = lambda x: x[0] == x[1]
# 使用.apply()和lambda函数遍历两列,检查元素相等性并创建新列
df['C'] = df[['A', 'B']].apply(check_identity, axis=1)
print(df)
输出:
A B C
0 1 1 True
1 2 3 False
2 3 3 True
3 4 5 False