pandas进行分组统计

最新推荐文章于 2024-08-20 22:19:11 发布

原创最新推荐文章于 2024-08-20 22:19:11 发布 · 4.1k 阅读

5 ·

CC 4.0 BY-SA版权

机器学习专栏收录该内容

7 篇文章

订阅专栏

博客介绍了获取foursquare中每个用户每个签到地点的频次的方法，可使用pandas的groupby方法进行统计，最后将userId、venueId和频次写入csv文件。

需要获取foursquare中每个用户每个签到地点的频次

可以用到pandas中groupby方法

data_ny.groupby(by=['userId','venueId']).size()

然后将userId ，venueId，频次写入csv中

userId = []
venueId = []
count = []
for i in data_gb:
    userId.append(i[0][0])
    venueId.append(i[0][1]) 
data_ub = pd.DataFrame()
data_ub['userId'] = userId
data_ub['venueId'] = venueId
data_ub['count'] = data_gb.size()
data_ub.to_csv('data/foursquare/data_ub.csv',index=None)