Remove double row column names from a pivot result

Question

I'd like to remove a 'double row' index header of a pivot result, so the following table:

Course_ID   CID-1   CID-2   CID-3
ID          
1           3.5     2.0     3.0
2           4.0     3.0     NaN

would look like this:

ID          CID-1   CID-2   CID-3           
1           3.5     2.0     3.0
2           4.0     3.0     NaN

How may I achieve this?

Here's the sample code:

sample = pd.DataFrame({'ID': [1, 1, 1, 2, 2], 
                       'Course_ID': ['CID-1', 'CID-2', 'CID-3', 'CID-1', 'CID-2'], 
                       'Grade': [3.5, 2, 3, 4, 3]})

result = pd.pivot_table(sample, index='ID', columns='Course_ID', values='Grade')

[very related to this](https://stackoverflow.com/questions/43756052/transform-pandas-pivot-table-to-regular-dataframe) — anky, Jun 16 '20 at 05:34
ah true, I couldn't find similar questions before I asked. Thank you @anky. Should i delete my question? — strivn, Jun 16 '20 at 05:45

FlorianGD · Answer 1 · 2020-06-16T11:03:15.623

7

You can do

result.columns.name = None

edited Jun 16 '20 at 11:03

answered Jun 16 '20 at 05:29

FlorianGD

2,336
1
15
32

score 0 · Accepted Answer · answered Jun 16 '20 at 05:24

Use DataFrame.rename_axis for remove columns name, here Course_ID and then DataFrame.reset_index for convert index to column ID:

result = result.rename_axis(None, axis=1).reset_index()
print (result)
   ID  CID-1  CID-2  CID-3
0   1    3.5    2.0    3.0
1   2    4.0    3.0    NaN

Remove double row column names from a pivot result

2 Answers2