Adding multiple different headers to df in python

Question

I have to read in multiple files and store these files as df in memory, since there are no headers in my csv file, I need to add multiple headers to each df manually. Each file has different headers.

What I did is using the if-elif statements, which works but very redundant. Does anyone have any idea how to make it more compact/smart?

if file_name == "20210101":
    headers = ["Name","Age","City"]
    df = df.read_csv(data, sep=";"dtype= str, names = headers)

elif file_name == "20210102":
    headers = ["Age","Location","Nation"]
    df = df.read_csv(data, sep=";"dtype= str, names = headers)

elif file_name == "20210103":
    headers = ["Account","Position","Sex"]
    df = df.read_csv(data, sep=";"dtype= str, names = headers)

timgeb · Accepted Answer · 2021-04-25 12:23:17Z

2

Use a dictionary mapping filenames to headers.

headers = {"20210101": ["Name","Age","City"], ...}
df = pd.read_csv(file_name, names=headers[file_name], ...)

answered Apr 25, 2021 at 12:23

timgeb

79.2k20 gold badges129 silver badges150 bronze badges

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

Collectives™ on Stack Overflow

Adding multiple different headers to df 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