python - 按列值设置多索引

Question

我有一个包含列的数据框，我想通过将相似值的条目分组在一列中来创建多索引

A a 3
B c 2
A d 1
A e 4
B a 5

变成了

A a 3
  d 1
  e 4
B c 2
  a 5

Answer 1

可以将索引设置为前两列，然后排序。

# create data
df = pd.DataFrame({'col1': {0: 'A', 1: 'B', 2: 'A', 3: 'A', 4: 'B'},
                   'col2': {0: 'a', 1: 'c', 2: 'd', 3: 'e', 4: 'a'},
                    'col3': {0: 3, 1: 2, 2: 1, 3: 4, 4: 5}})
# set and sort index
df.set_index(['col1', 'col2']).sort_index(level=['col1', 'col2'])

输出

           col3
col1 col2      
A    a        3
     d        1
     e        4
B    a        5
     c        2