Copy a Multi-Index column of a Pandas Dataframe including the second header

Question

Background - I have a dataset with two headers that I've read from a CSV...

df = pd.read_csv(file, header=[0,1])
print(df)

     A   B   C
     a   b   c
---------------
0    1   2   3
1    4   5   6
2    7   8   9
       ...

What I Want - I'd like to duplicate one of the columns (say, A) into a new column D, such that...

     A   B   C   D
     a   b   c   a
----------------------
0    1   2   3   1
1    4   5   6   4
2    7   8   9   7
       ...

What I'm Getting - ...but when I do this using df['D'] = df['A'], what I get is a copy without the content of the second header...

     A   B   C   D
     a   b   c   
----------------------
0    1   2   3   1
1    4   5   6   4
2    7   8   9   7
       ...

TLDR - How can I capture the content of the second header of my multi-index dataframe when copying a column?

cottontail · Accepted Answer · 2022-06-16 22:37:10Z

1

Need to specify both levels of multiindex

# assign ('A', 'a') values to ('D', 'a') column
df[('D','a')] = df[('A','a')] # or df['D','a'] = df['A']

answered Jun 16, 2022 at 22:37

cottontail

25.7k25 gold badges187 silver badges177 bronze badges

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

1 Answer 1