python的dataframe数据分组

def test4():
    df = pd.DataFrame({'key1': ['a', 'a', 'b', 'b', 'a'],
                       'key2':['one', 'two', 'one', 'two', 'one'],
                       'data1':np.random.randn(5),
                       'data2':np.random.randn(5)})
    print df
    df_g = df.groupby(['key1'])
    print len(df_g.size())
    for name, group in df_g:
        print name
        print group