Pandas - How to flatten a hierarchical index in columns

后端 未结 17 1443
忘掉有多难
忘掉有多难 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:14

    Following @jxstanford and @tvt173, I wrote a quick function which should do the trick, regardless of string/int column names:

    def flatten_cols(df):
        df.columns = [
            '_'.join(tuple(map(str, t))).rstrip('_') 
            for t in df.columns.values
            ]
        return df
    

提交回复
热议问题