How to merge two columns of a pivot_table from pandas in python?

Question

I obtained a dataframe using pd.pivot_table, that looks like this:

    foo      bar      
Cond1     60   65    60    65
Cond2                      
50        200  210  16.7  15.2
100       200  210  14.9  13.5

I need to get an output that looks like this by merging the foo and bar columns:

           foo(bar)      
    Cond1     60          65
    Cond2                      
    50        200(16.7)  210(15.2)
    100       200(14.9)  210(13.5)

Is this possible in python using only pandas or numpy or internal libraries?

jezrael · Accepted Answer · 2021-11-15 07:49:54Z

0

Solution for no MultiIndex in ouput with DataFrame.xs, casting to strings and last join by +:

df1 = df.xs('foo', axis=1, level=0).astype(str)
df2 = df.xs('bar', axis=1, level=0).astype(str)

df = df1 + '(' + df2 + ')'

Solution with MultiIndex:

df1 = df.xs('foo', axis=1, level=0, drop_level=False).astype(str)
df2 = df.xs('bar', axis=1, level=0, drop_level=False).astype(str)

df = df1.rename(columns={'foo':'foo(bar)'})+'('+ df2.rename(columns={'bar':'foo(bar)'})+ ')'

answered Nov 15, 2021 at 7:49

jezrael

868k103 gold badges1.4k silver badges1.3k bronze badges

Sign up to request clarification or add additional context in comments.

Collectives™ on Stack Overflow

How to merge two columns of a pivot_table from pandas in python?

1 Answer 1

Comments

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

Comments

Your Answer

Sign up or log in

Post as a guest

Related