python - 如何对一级索引进行分组并将函数应用于 Pandas 中的二级索引

Question

我有这样的多级数据框“df”:

             col1 col2
first second
a        0    5    5
         1    5    5
         2    5    5
b        0    5    5
         1    5    5

我想将函数 func (exp: 'lambda x: x*10') 应用到 second，有点像:

df.groupby(level='first').second.apply(func)

结果会像这样:

             col1 col2
first second
a        0    5    5
         10   5    5
         20   5    5
b        0    5    5
         10   5    5

上述命令不适用于 second 不是列，因此 .second 不被 Pandas 接受。

我不想通过 df.reset_index() 来做到这一点，blablabla...，然后最后是 df.set_index()。 我更愿意这样做在一个命令中，怎么做？

Answer 1

您可以使用索引的 set_levels 方法来更改给定级别的值。所以对于给定的 func 和 level 你可以这样做:

new_values = map(func, df.index.get_level_values(level))
df.index.set_levels(new_values, level, inplace=True)