How to efficiently create a pivot table?

前端 未结 2 1344
星月不相逢
星月不相逢 2021-01-07 06:54

I do have a dataframe like this:

import pandas as pd

df = pd.DataFrame({\"c0\": list(\'ABC\'),
                   \"c1\": [\" \".join(list(\'ab\')), \" \".j         


        
2条回答
  •  没有蜡笔的小新
    2021-01-07 07:00

    Option 1

    import numpy as np, pandas as pd
    
    s = df.c1.str.split()
    l = s.str.len()
    newdf = df.loc[df.index.repeat(l)].assign(c1=np.concatenate(s)).set_index(['c0', 'c1'])
    newdf
    
          c2
    c0 c1   
    A  a   D
       b   D
    B  d   E
       e   E
       f   E
    C  s   F
    

    Option 2
    Should be faster

    import numpy as np, pandas as pd
    
    s = np.core.defchararray.split(df.c1.values.astype(str), ' ')
    l = [len(x) for x in s.tolist()]
    r = np.arange(len(s)).repeat(l)
    i = pd.MultiIndex.from_arrays([
        df.c0.values[r],
        np.concatenate(s)
    ], names=['c0', 'c1'])
    newdf = pd.DataFrame({'c2': df.c2.values[r]}, i)
    newdf
    
          c2
    c0 c1   
    A  a   D
       b   D
    B  d   E
       e   E
       f   E
    C  s   F
    

提交回复
热议问题