我正在尝试将简单的函数应用于 pandas 中的组。我有这个数据框,我可以根据它进行分组type
:
df = pandas.DataFrame({"id": ["a", "b", "c", "d"], "v": [1,2,3,4], "type": ["X", "Y", "Y", "Y"]}).set_index("id")
df.groupby("type").mean() # gets the mean per type
我想应用像这样的函数np.log2
在取每组的平均值之前仅对组进行计算。这不起作用,因为apply
是元素明智的并且type
(以及潜在的其他列df
在真实场景中)不是数字:
# fails
df.apply(np.log2).groupby("type").mean()
有办法申请吗np.log2
只对取平均值之前的组进行?我想transform
这将是答案,但问题是它返回一个没有原始数据的数据框type
列:
df.groupby("type").transform(np.log2)
v
id
a 0.000000
b 1.000000
c 1.584963
d 2.000000
像分组然后应用这样的变体不起作用:df.groupby("type").apply(np.log2)
。这样做的正确方法是什么?