首页
学习
活动
专区
工具
TVP
发布
精选内容/技术社群/优惠产品,尽在小程序
立即前往

Python数据分析之groupby语法糖对分组进行迭代语法糖一:选取一个或多个列

对于dataframe的groupby聚合函数来说,我们适当了解下语法糖,会对数据分析起到事半功倍的效果。

对分组进行迭代

首先看下各字段的类型

代码语言:javascript
复制
import numpy as np
import pandas as pd
import pymysql
conn = pymysql.connect(host='localhost', user='root', passwd='123456', db='test', port=3306, charset='utf8')
jianshu = pd.read_sql('select * from jianshu1',conn)
jianshu.dtypes

可以看出,view这些字段是整数类型的数据,但这里是object数据,所以我们需要进行数据类型的修改,以view为例。

代码语言:javascript
复制
jianshu['view'] = jianshu['view'].astype('int64')
jianshu.dtypes

把user列作为索引。

代码语言:javascript
复制
jianshu.set_index('user',inplace=True)
jianshu

通过分组后的数据类型为groupby对象,可进行迭代。

代码语言:javascript
复制
jianshu.groupby(jianshu.index)
代码语言:javascript
复制
for name,group in jianshu.groupby(jianshu.index):
    print(name,group)

语法糖一:选取一个或多个列

代码语言:javascript
复制
jianshu.groupby(jianshu.index)[['view']].sum()

语法糖二:数据聚合 通过aggregate或者agg方法

代码语言:javascript
复制
jianshu.groupby(jianshu.index)[['view']].agg(['mean','sum'])
下一篇
举报
领券