Pandas - How to flatten a hierarchical index in columns

后端 未结 17 1092
忘掉有多难
忘掉有多难 2020-11-22 02:55

I have a data frame with a hierarchical index in axis 1 (columns) (from a groupby.agg operation):

     USAF   WBAN  year  month  day  s_PC  s_CL         


        
17条回答
  •  时光说笑
    2020-11-22 03:23

    Another simple routine.

    def flatten_columns(df, sep='.'):
        def _remove_empty(column_name):
            return tuple(element for element in column_name if element)
        def _join(column_name):
            return sep.join(column_name)
    
        new_columns = [_join(_remove_empty(column)) for column in df.columns.values]
        df.columns = new_columns
    

提交回复
热议问题