python - 将函数应用于两列 Pandas

Question

我正在尝试应用一个函数来计算 pandas 数据框的两列的 R2 分数。如果首先我按其他列分组，它会起作用，但是当应用于整个数据框时，它会失败。

这是有效的代码:

from sklearn.metrics import r2_score
df = pd.DataFrame({'cat':['A','A','B','B'], 'num1':[.1,.2,.3,.4], 
'num2': [.1,.2,.3,.4]})

df.groupby('cat').apply(lambda x: r2_score(x['num1'],x['num2']))

有了这个输出:

cat
A    1.0
B    1.0
dtype: float64

但这不起作用:

df.apply(lambda x: r2_score(x['num1'],x['num2']))

出现关键错误: KeyError: ('num1', '发生在索引 cat')

有人可以解释为什么会出现这种情况，阐明 apply 函数的内部工作原理，并为后者 apply 提供解决方案吗？

Answer 1

我相信您需要传递列才能发挥作用:

a = r2_score(df['num1'],df['num2'])
print (a)
1.0