python中Groupby使用（四）-通过函数进行分组

#-*- coding:utf-8 -*-
import pandas as pd
import numpy as np
people=pd.DataFrame(np.random.randn(5,5),columns=['a','b','c','d','e']
                    ,index=['Joe','Steve','Wes','Jim','Travis'])
print people.groupby(len).sum()#如果希望根据人名长度进行分组，虽然可以求取一个字符串长度数组，
#但其实仅仅传入len函数就可以了

#将函数跟数组、列表、字典、Series混合使用也不是问题，因为任何东西最终都会被转换为数组
key_list=['one','one','one','two','two']
print people.groupby([len,key_list]).sum()