我有一個代表顧客登記（訪問）餐廳的數據框。year簡直就是在餐廳辦理入住的那一年。average_checkin我想要做的是在我的初始數據框中添加一列df，表示每年餐廳的平均訪問次數。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'])# here i count the total number of checkins a restaurant hadd = df.groupby('restaurant_id')['year'].count().to_dict()df['nb_checkin'] = df['restaurant_id'].map(d)mean_checkin= df.groupby(['restaurant_id','year']).agg({'nb_checkin':[np.mean]})mean_checkin.columns = ['mean_checkin']mean_checkin.reset_index()# the values in mean_checkin makes no sens#I need to merge it with df to add that new column我對 pandas 庫還是個新手，我嘗試過類似的方法，但我的結果毫無意義。我的語法有問題嗎？如果需要任何澄清，請詢問。
1 回答
MMMHUHU
TA貢獻1834條經驗 獲得超8個贊
每年的平均訪問次數可以通過餐廳的總訪問次數除以您擁有數據的唯一年份數來計算。
grouped = df.groupby(["restaurant_id"])
avg_annual_visits = grouped["year"].count() / grouped["year"].nunique()
avg_annual_visits = avg_annual_visits.rename("avg_annual_visits")
print(avg_annual_visits)
restaurant_id
--1UhMGODdWsrMastO9DZw 3.0
--6MefnULPED_I942VcFNA 2.0
Name: avg_annual_visits, dtype: float64
然后，如果您想將其合并回原始數據：
df = df.merge(avg_annual_visits, left_on="restaurant_id", right_index=True)
print(df)
restaurant_id year avg_annual_visits
0 --1UhMGODdWsrMastO9DZw 2016 3.0
1 --1UhMGODdWsrMastO9DZw 2016 3.0
2 --1UhMGODdWsrMastO9DZw 2016 3.0
3 --1UhMGODdWsrMastO9DZw 2016 3.0
4 --1UhMGODdWsrMastO9DZw 2017 3.0
5 --1UhMGODdWsrMastO9DZw 2017 3.0
6 --6MefnULPED_I942VcFNA 2011 2.0
7 --6MefnULPED_I942VcFNA 2011 2.0
8 --6MefnULPED_I942VcFNA 2012 2.0
9 --6MefnULPED_I942VcFNA 2012 2.0
添加回答
舉報
0/150
提交
取消