python - Pandas 意外的 set_index 行为

Question

data = [['g1','a',1],['g1','b',2],['g2','b',3],['g2','a',4]]
df = pandas.DataFrame(data=data, columns=['group','name','count'])
print df.set_index(['group','name'])
print df.set_index(['name','group'])

            count
group name       
g1    a         1
      b         2
g2    b         3
      a         4
            count
name group       
a    g1         1
b    g1         2
     g2         3
a    g2         4

这种行为让我很惊讶，因为我期待第二个输出像

            count
name group       
a    g1         1
     g2         4
b    g1         2
     g2         3

Answer 1

首先需要对 DataFrame 进行排序以获得所需的输出:

In [12]: df.sort_values('name').set_index(['name','group'])
Out[12]: 
            count
name group       
a    g1         1
     g2         4
b    g1         2
     g2         3