英文原文: 06 - Lesson
我们看一下 groupby 这个函数。
import pandas as pd
import sys
print('Python version ' + sys.version)
print('Pandas version ' + pd.__version__)
Python version 3.6.1 | packaged by conda-forge | (default, Mar 23 2017, 21:57:00)
[GCC 4.2.1 Compatible Apple LLVM 6.1.0 (clang-602.0.53)]
Pandas version 0.19.2
d = {'one':[1,1,1,1,1],
'two':[2,2,2,2,2],
'letter':['a','a','b','b','c']}
df = pd.DataFrame(d)
df
| letter | one | two |
---|
0 | a | 1 | 2 |
---|
1 | a | 1 | 2 |
---|
2 | b | 1 | 2 |
---|
3 | b | 1 | 2 |
---|
4 | c | 1 | 2 |
---|
one = df.groupby('letter')
one.sum()
letterone = df.groupby(['letter','one']).sum()
letterone
letterone.index
MultiIndex(levels=[['a', 'b', 'c'], [1]],
labels=[[0, 1, 2], [0, 0, 0]],
names=['letter', 'one'])
你可能不想把用来分组的列名字作为索引,像下面的做法很容易实现。
letterone = df.groupby(['letter','one'], as_index=False).sum()
letterone
letterone.index
Int64Index([0, 1, 2], dtype='int64')