Pandas DataFrame中删除列级的方法链接解决方案

问题描述

我在重塑和查询我在 pandas DataFrames中的数据时使用的是Lot of方法链。有时会为in索引(行)和列创建额外的和不必要的级别。如果是，例如在索引(行轴)上，可以使用DataFrame.reset_index()：

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

reset_index函数允许用户继续链接方法并继续使用DataFrame。

尽管如此，我从来没有为Column_Axis找到一个等价的解决方案。有吗？

推荐答案

您可以只调用stack列(将其移动到索引中)，并使用Drop=True调用reset_index，或者您可以使用reset_index()作为起点编写reset_columns()方法(请参阅Frame.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)

替代方案：猴贴溶液

def drop_column_levels(self, level=None, inplace=False):
        """
        For DataFrame with multi-level columns, drops one or more levels.
        For a standard index, or if dropping all levels of the MultiIndex, will revert
        back to using a classic RangeIndexer for column names.

        Parameters
        ----------
        level : int, str, tuple, or list, default None
            Only remove the given levels from the index. Removes all levels by
            default
        inplace : boolean, default False
            Modify the DataFrame in place (do not create a new object)

        Returns
        -------
        resetted : DataFrame
        """
        if inplace:
            new_obj = self
        else:
            new_obj = self.copy()

        new_columns = pd.core.common._default_index(len(new_obj.columns))
        if isinstance(self.index, pd.MultiIndex):
            if level is not None:
                if not isinstance(level, (tuple, list)):
                    level = [level]
                level = [self.index._get_level_number(lev) for lev in level]
                if len(level) < len(self.columns.levels):
                    new_columns = self.columns.droplevel(level)

        new_obj.columns = new_columns
        if not inplace:
            return new_obj

# Monkey patch the DataFrame class
pd.DataFrame.drop_column_levels = drop_column_levels