比较.apply() pandas中的多列

df.apply(func, axis=0)

import pandas as pd

# 创建示例DataFrame
data = {'col1': [1, 2, 3, 4, 5],
        'col2': [2, 4, 1, 5, 3]}
df = pd.DataFrame(data)

# 自定义函数，比较两列数据的大小
def compare_cols(row):
    if row['col1'] > row['col2']:
        return 'col1 > col2'
    elif row['col1'] < row['col2']:
        return 'col1 < col2'
    else:
        return 'col1 = col2'

# 应用自定义函数到多列数据
df['compare_result'] = df.apply(compare_cols, axis=1)

print(df)

   col1  col2 compare_result
0     1     2    col1 < col2
1     2     4    col1 < col2
2     3     1    col1 > col2
3     4     5    col1 < col2
4     5     3    col1 > col2