Keyerror showing while using python pandas drop function

Question

import pandas as pd
import numpy as np
from sklearn.impute import SimpleImputer
df = pd.read_csv("Covid-19 Global Data.csv")

df.head(3)

      Date_reported   Country_code    Country        WHO_region     New_cases                New_deaths
     0     03-01-20          AF       Afghanistan      EMRO           0                               0                                   
     1     04-01-20          AF       Afghanistan      EMRO           0                               0                                   
     2     05-01-20          AF       Afghanistan      EMRO           0                               0                                  

df.drop(["Country"],axis=1,inplace=True)

Showing keyerror everytime. The dataframe is constructed perfectly but KeyError is popping up.

Grayrigel · Accepted Answer · 2020-10-25 14:50:28Z

1

The error could be due to the additional whitespace in the column name. Perhaps try adding a whitespace and dropping it:

df.drop(["Country "],axis=1,inplace=True)

Or

df.drop([" Country"],axis=1,inplace=True)
# df.drop([" Country "],axis=1,inplace=True)

One better way would be strip extra whitespace from the column names with the following:

df.columns = df.columns.str.strip()
df.drop(["Country"],axis=1,inplace=True)

edited Oct 25, 2020 at 14:50

answered Oct 25, 2020 at 14:45

Grayrigel

3,6045 gold badges18 silver badges36 bronze badges

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

Comments

Soaib · Accepted Answer · 2020-10-25 14:31:09Z

1

Seems like you have already dropped the Country column.

Worked well for me.

Restart the karnel and try to run it again.

answered Oct 25, 2020 at 14:31

Soaib

9611 bronze badges

2 Comments

sam700 Over a year ago

No i didn't drop any of the columns of the data frame. The code is not working for any of the columns even after restarting the kernel

Soaib Over a year ago

Then check the whitespace of the Country name. Like ' Country' or 'Country '

MarianD · Accepted Answer · 2020-10-25 23:01:28Z

1

Use

print(df.columns)

to see real names of your columns. You obtained something as

Index(['Date_reported', 'Country_code', 'Country', 'WHO_region', 'New_cases',
       'New_deaths'],
      dtype='object')

edited Oct 25, 2020 at 23:01

answered Oct 25, 2020 at 14:46

MarianD

14.4k12 gold badges50 silver badges61 bronze badges

Collectives™ on Stack Overflow

Keyerror showing while using python pandas drop function

3 Answers 3

Comments

2 Comments

Comments

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

3 Answers 3

Comments

2 Comments

Comments

Your Answer

Sign up or log in

Post as a guest

Related