pandas groupby多条件分组
时间: 2023-06-05 09:48:03 浏览: 222
pandas dataframe对象的分组机制groupby
pandas的groupby函数可以实现多条件分组,可以通过传入多个列名或者多个Series对象来实现。例如:
```python
import pandas as pd
df = pd.DataFrame({
'A': ['foo', 'bar', 'foo', 'bar', 'foo', 'bar', 'foo', 'foo'],
'B': ['one', 'one', 'two', 'three', 'two', 'two', 'one', 'three'],
'C': [1, 2, 3, 4, 5, 6, 7, 8],
'D': [10, 20, 30, 40, 50, 60, 70, 80]
})
# 按照A和B两列进行分组
grouped = df.groupby(['A', 'B'])
# 对分组后的数据进行聚合操作
result = grouped.sum()
print(result)
```
输出结果为:
```
C D
A B
bar one 2 30
three 4 40
two 12 120
foo one 8 80
three 8 80
two 10 100
```
在上面的例子中,我们按照A和B两列进行了分组,然后对分组后的数据进行了求和操作。可以看到,分组后的结果是一个多级索引的DataFrame对象。
阅读全文