Adding a header/another row to a pandas dataframe based on some columns

Question

I am trying to add a subheader to a pandas data frame based on some attributes. an example is the following:

Current input (already a dataframe with data generated):

Ideal output: Currently, I have all of the attributes in the second row but would like to add a header/above row based on a grouping of certain attributes.

Note: The main key is the order ID not sure if that helps.

SomeDude · Accepted Answer · 2022-08-24 22:20:54Z

1

I assume you already have a dataframe with the column names in the inner level shown in your picture. Then one way to do without knowing the column names is:

cols = df.columns
df.columns = pd.MultiIndex.from_arrays(
    [[''] * len(cols), cols]).map(lambda x: (x[1].split()[0] + ' details', x[1]))

edited Aug 24, 2022 at 22:20

answered Aug 24, 2022 at 22:07

SomeDude

14.3k5 gold badges26 silver badges49 bronze badges

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

4 Comments

SOScodehelp Over a year ago

Thank you very much and yes I have a df. but instead of having it with the details, is there a way to manually add in the column names? I have just showed an example in the question, but I actually have around 40 columns and not all of the grouping ends in "details".

SomeDude Over a year ago

Yes if you give the right input and right output, better answer can be provided.

SOScodehelp Over a year ago

I have updated. Please let me know if I could provide more clarity

SomeDude Over a year ago

Do you have a mapping(sort of a dict) like {'SKU Details': ['Product Name', 'SKU Code', 'SKU Expiration Date']} ?

Michael Hodel · Accepted Answer · 2022-08-24 21:26:51Z

0

import pandas as pd

cols = pd.MultiIndex.from_tuples([("Buyer details", "Buyer Name"),
                                  ("Buyer details", "Buyer Address"),
                                  ("Oder Details", "Order Id"),
                                  ("Oder Details", "Order description"),
                                  ("Oder Details", "Order person"),
                                  ("Item Details", "Item Link"),
                                  ("Item Details", "Item cost")])
df = pd.DataFrame(columns=cols)
print(df)

answered Aug 24, 2022 at 21:26

Michael Hodel

3,0381 gold badge8 silver badges12 bronze badges

1 Comment

SOScodehelp Over a year ago

Thank you, I have an existing data frame with columns of the second header, and want to group based on the first. is there a way to do that?

Collectives™ on Stack Overflow

Adding a header/another row to a pandas dataframe based on some columns

2 Answers 2

4 Comments

1 Comment

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

2 Answers 2

4 Comments

1 Comment

Your Answer

Sign up or log in

Post as a guest

Related