python - 按组计算几何平均值

Question

我有一个包含四列的数据集，类似于下面显示的前四列。我想添加另一列，显示某些组的“价格”中值的几何平均值，其中每个组由“类型”列确定。
我怎样才能做到这一点？结果将是类似于下面标记为“geomean_price_bytype”的列。

Answer 1

使用 GroupBy.transform 与 gmean :

from scipy.stats.mstats import gmean

#if necessary remove `,` and `$`
#df['price'] = df['price'].str.lstrip('$').str.replace(',', '').astype(int)


df['new'] = df.groupby('type')['price'].transform(gmean)

或自定义 lambda 函数:

gmean1 = lambda x: x.product() ** (1 / float(len(x)))
df['new'] = df.groupby('type')['price'].transform(gmean1)