How to pass my variable as a string into function

Question

I have wrote lines of code to convert items of a pandas dataframe column to a dict with appropriate values. The code looks like this:

states = df.STATE.unique()
states.sort()
states_values = [i for i in range(0,len(states))]
state_dict = {states[i]: states_values[i] for i in range(len(states))}

But I have many columns to convert so I tried using a function and wrote a function like this

def make_dict(dataframe,column_name):
    keys = dataframe.column_name.unique()
    keys.sort()
    values = [i for i in range(0,len(keys))]
    dict = {keys[i]: values[i] for i in range(len(keys))}
    return dict

But When I tried

city_dict = make_dict(df,'CITY')

Where CITY is my column name, it throws an error

AttributeError: 'DataFrame' object has no attribute 'column_name'

I have tried setting a variable and did it but same error.

tried with df['CITY'] over df.CITY but still doesn't work out. Please let me know what am I missing.

John Gordon · Accepted Answer · 2022-06-22 00:25:15Z

1

You can use getattr() to fetch arbitrary attributes.

keys = getattr(dataframe, column_name).unique()

answered Jun 22, 2022 at 0:25

John Gordon

33.8k9 gold badges48 silver badges72 bronze badges

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

2 Comments

Sharp Thwey Thit Over a year ago

Thank you so much @John, This solved everything. Thank you!!

Sharp Thwey Thit Over a year ago

Sorry for that I couldn't upvote your answer as my reputation is too low!!

Collectives™ on Stack Overflow

How to pass my variable as a string into function

1 Answer 1

2 Comments

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

2 Comments

Your Answer

Sign up or log in

Post as a guest

Related