KeyError using apply function

Question

I have a data frame with the following columns, and I'm simply trying to add a new column by transforming an existing one. I don't understand why I get this error, especially given is that the data frame is fine and I can groupby on Zip without any index problems.

print(df.columns)

# Index(['First Col', 'Year', 'Submitted', 'Allowed', 'Provided', 'X', 'Zip'],
# dtype='object')

print(df['Zip'])

# 0       10523
# 1       11803
# 2       22939
# 3       21742
# 4       21801
# 5       21804
# ...

df['NEW'] = df.apply(lambda row: cool_fn(row['Zip']))
KeyError: ('Zip', 'occurred at index First Col')

jezrael · Accepted Answer · 2019-05-20 08:27:04Z

3

For processing per rows is necessary add axis=1 to DataFrame.apply:

df['NEW'] = df.apply(lambda row: cool_fn(row['Zip']), axis=1)

Or use Series.apply, then lambda should be omit:

df['NEW'] = df['Zip'].apply(cool_fn)

answered May 20, 2019 at 8:27

jezrael

868k103 gold badges1.4k silver badges1.3k bronze badges

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

Collectives™ on Stack Overflow

KeyError using apply function

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