将 Apply 函数结果添加到原始 df

Question

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

我想将下面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')