Применить функцию к двум столбцам панд

Я пытаюсь применить функцию, которая вычисляет баллы R2 до двух столбцов кадра данных pandas. Он работает, если сначала я группирую какой-либо другой столбец, но когда применяется ко всему кадру данных, он терпит неудачу.

Это код, который работает:

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 и предоставляя решение для последних?