每组的新变量

Question

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

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

不是聚合！这是一个新的变量，每组都不一样。 我尝试了熊猫的一切-“分组方式”想要聚合或分割我的数据！我只想计算新变量。你知道吗

Answer 1

我们可以试试

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')

eh3数据<

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