DataFrame groupby
时间: 2023-08-15 17:33:02 浏览: 103
Pandas DataFrame Notes
Pandas 中的 `groupby()` 函数可以对 DataFrame 进行分组操作,类似于 SQL 中的 GROUP BY。它的基本语法如下:
```python
df.groupby(by=None, axis=0, level=None, as_index=True, sort=True, group_keys=True, squeeze=False, observed=False, **kwargs)
```
其中,参数的含义如下:
- `by`: 分组依据,可以是列名、列标签或者列标签的列表。
- `axis`: 分组轴的方向,0 表示按行分组,1 表示按列分组,默认为 0。
- `level`: 分组时使用的层级,用于处理多层索引的 DataFrame。
- `as_index`: 是否将分组依据作为索引,如果为 True,则分组依据将成为 DataFrame 的索引。
- `sort`: 是否对分组结果进行排序,默认为 True。
- `group_keys`: 是否在结果中包含分组键,默认为 True。
- `squeeze`: 是否对结果进行压缩,如果分组后的结果只有一列,是否将其转换为 Series,默认为 False。
- `observed`: 是否包含所有的观测值,如果为 True,则只包含出现在分组键中的观测值,否则包含所有观测值。
下面是一个简单的例子,假设有一个 DataFrame 包含了几个班级学生的成绩信息:
```python
import pandas as pd
df = pd.DataFrame({
'class': ['A', 'A', 'B', 'B', 'B', 'C', 'C', 'C'],
'name': ['Alice', 'Bob', 'Charlie', 'Dave', 'Eve', 'Frank', 'Grace', 'Henry'],
'score': [80, 75, 90, 85, 80, 95, 90, 85]
})
```
可以按照班级进行分组,统计每个班级的平均成绩:
```python
grouped = df.groupby('class')
result = grouped.mean()
print(result)
```
输出结果为:
```
score
class
A 77.500000
B 85.000000
C 90.000000
```
也可以按照多个列进行分组,例如按照班级和成绩进行分组:
```python
grouped = df.groupby(['class', 'score'])
result = grouped.mean()
print(result)
```
输出结果为:
```
name
class score
A 75 Bob
80 Alice
B 80 Eve
85 Dave
90 Charlie
C 85 Henry
90 Grace
95 Frank
```
在分组后,可以使用 `agg()` 函数对分组后的结果进行聚合操作,例如计算每个班级的总分:
```python
result = grouped.agg({'score': 'sum'})
print(result)
```
输出结果为:
```
score
class
A 155
B 255
C 270
```
阅读全文