python - 替代 set_level_values

Question

我有一系列的形式

  groupid     period    
        1    2011-02     5
        1    2011-03     6
        1    2011-04     7
        1    2011-05     8

我想将句点向后移动一个句点。我可以使用

创建新索引

df.index.get_level_values('period').shift(-1)

但后来我不知道如何强制他们返回。set_level_values() 不存在。如何覆盖 MultiIndex 的第二级？

Answer 1

您可以使用 MultiIndex.set_levels :

print (df.index)
MultiIndex(levels=[[1], [2011-02, 2011-03, 2011-04, 2011-05]],
           labels=[[0, 0, 0, 0], [0, 1, 2, 3]],
           names=['groupid', 'period'])

print (df.index.get_level_values('period').shift(-1))
PeriodIndex(['2011-01', '2011-02', '2011-03', '2011-04'], 
            dtype='period[M]', name='period', freq='M')

df.index.set_levels([df.index.get_level_values('groupid'), 
                     df.index.get_level_values('period').shift(-1)], inplace=True)
print (df)
groupid  period 
1        2011-01    5
         2011-02    6
         2011-03    7
         2011-04    8
Name: val, dtype: int64