Python 每组的新变量-熊猫

我爱R，但现在我需要熊猫。 在R中，我可以做：

data %>% group_by(sym) %>% mutate(s = mean(price))

这不是聚合！这是一个新的变量，每个组都不同。 我在pandas-group中尝试了一切，希望聚合或拆分我的数据！我只想计算新变量。

我们可以试试

import pandas as pd
data['s'] = data['price'].groupby(data['sym']).transform('mean')
print(data)
#   price sym           s
#0    125   A  129.000000
#1    133   A  129.000000
#2     50   B   77.333333
#3     62   B   77.333333
#4    120   B   77.333333

或者像@MaxU提到的那样

data['s'] = data.groupby('sym')['price'].transform('mean')

数据

---

不幸的是，并非所有pandas专家都具有与R相同的专业水平。因此，如果您可以使用大约5行的示例输入数据集/数据帧和所需的输出/结果数据集，则可以更轻松地为您提供帮助…我会将其简化一点：data.groupby'sym'['price'].transform'mean'

data = pd.DataFrame({'sym': ['A', 'A', 'B', 'B', 'B'], 'price': [125, 133, 50, 62, 120]})