Добавление результатов функции Apply в исходный df

Question

df
date         fruit      var1    var2
2010-01-01    apple      40      30
2010-01-01    banana     40      30
...

Я хочу добавить в качестве нового столбца с именем avg результаты приведенной ниже функции apply к исходному df, в идеале без создания переменной с именем test.

test = df.groupby('date').apply(some_function, "var1", "var2")
print(test)
date
2010-01-01    0.744337
2010-01-02    0.7445

Ожидается Вывод:

date         fruit      var1    var2      avg
2010-01-01    apple      40      30     0.744337
2010-01-01    banana     40      30     0.744337
...

Answer 1

Если невозможно изменить использование функции:

df = df.join(test.rename('avg'), on='date')