pandas用列表进行分组

people = pd.DataFrame(np.random.randn(5, 5),
                      columns=['a', 'b', 'c', 'd', 'e'],
                      index=['Joe', 'Steve', 'Wes', 'Jim', 'Travis'])
people.iloc[2:3, [1, 2]] = np.nan # Add a few NA values
people

               a 	        b 	       c 	       d 	       e
Joe 	1.007189 	-1.296221 	0.274992 	0.228913 	1.352917
Steve 	0.886429 	-2.001637 	-0.371843 	1.669025 	-0.438570
Wes 	-0.539741 	NaN 	         NaN 	-1.021228 	-0.577087
Jim 	0.124121 	0.302614 	0.523772 	0.000940 	1.343810
Travis 	-0.713544 	-0.831154 	-2.370232 	-1.860761 	-0.860757

key_list = ['one', 'one', 'one', 'two', 'two']
what = people.groupby(key_list)
for i in what:
    print(i)

('one',               a         b         c         d         e
         Joe    1.007189 -1.296221  0.274992  0.228913  1.352917
         Steve  0.886429 -2.001637 -0.371843  1.669025 -0.438570
         Wes   -0.539741       NaN       NaN -1.021228 -0.577087)
('two',                a         b         c         d         e
         Jim     0.124121  0.302614  0.523772  0.000940  1.343810
         Travis -0.713544 -0.831154 -2.370232 -1.860761 -0.860757)

groupby默认对行方向进行分组，所以把列表每个元素对应一行，进行分组。