pandas数据帧中降低列级别的方法链接解决方案

Question

在重塑和查询pandasDataFrames中的数据时，我使用了大量的方法链接。有时会为索引（行）和列创建额外的和不必要的级别。如果是这样，例如在索引（行轴）上，这很容易通过使用DataFrame.reset_index()来解决：

df.query('some query')
   .apply(cool_func)
   .reset_index('unwanted_index_level',drop=True) # <====
   .apply(another_cool_func)

reset_index函数允许继续使用链式方法并继续使用DataFrame。在

然而，我从来没有找到一个与列轴相等的解决方案。有没有？在

Answer 1

您只需stack列（将其移动到索引）并使用drop=True调用reset_index，或者可以使用reset_index()作为起点编写reset_columns()方法（请参见框架py#L2940)在

df.query('some query')
   .apply(cool_func)
   .stack(level='unwanted_col_level_name')
   .reset_index('unwanted_col_level_name',drop=True)
   .apply(another_cool_func)

---

替代方案：猴头贴解决方案

^{pr2}$