熊猫：计算每"year"列总值的标准

Question

我有一个数据框架，表示顾客在餐馆的登记(访问(。year只是一家餐厅办理入住手续的年份。

• 我想做的是在我的初始数据框df中添加一列std_checkin，它表示每年访问的标准偏差。因此，我需要计算每年总访问量的标准差

data = {
'restaurant_id':  ['--1UhMGODdWsrMastO9DZw', '--1UhMGODdWsrMastO9DZw','--1UhMGODdWsrMastO9DZw','--1UhMGODdWsrMastO9DZw','--1UhMGODdWsrMastO9DZw','--1UhMGODdWsrMastO9DZw','--6MefnULPED_I942VcFNA','--6MefnULPED_I942VcFNA','--6MefnULPED_I942VcFNA','--6MefnULPED_I942VcFNA'],
'year': ['2016','2016','2016','2016','2017','2017','2011','2011','2012','2012'],
}
df = pd.DataFrame (data, columns = ['restaurant_id','year'])
# total number of checkins per restaurant
d = df.groupby('restaurant_id')['year'].count().to_dict()
df['nb_checkin'] = df['restaurant_id'].map(d)

grouped = df.groupby(["restaurant_id"])
avg_annual_visits = grouped["year"].count() / grouped["year"].nunique()
avg_annual_visits = avg_annual_visits.rename("avg_annual_visits")
df = df.merge(avg_annual_visits, left_on="restaurant_id", right_index=True)
df.head(10)

从这里开始，我不知道如何用熊猫写出我想要的东西。如果需要任何澄清，请询问。

谢谢！

Answer 1

我想你想做的是：

counts = df.groupby('restaurant_id')['year'].value_counts()
counts.std(level='restaurant_id')

counts的输出，即每家餐厅每年的总访问量：

restaurant_id           year
--1UhMGODdWsrMastO9DZw  2016    4
2017    2
--6MefnULPED_I942VcFNA  2011    2
2012    2
Name: year, dtype: int64

和std的输出

restaurant_id
--1UhMGODdWsrMastO9DZw    1.414214
--6MefnULPED_I942VcFNA    0.000000
Name: year, dtype: float64