How to delete rows in a data frame with for loop

Question

So I want to delete some rows based on a condition.

I tried to drop as it is explained in the website.

My dataframe looks like this:

jobs
   job_name  number
0  job1      163
1  job2      200
2  job3      3

And I have a list:

my_jobs = [job2, job3]

My condition:

for job in jobs['job_name']:
    if job not in my_jobs:
        jobs_df.drop(job, axis=0)

And I want my dataframe looks like this:

jobs
   job_name  number
1  job2      200
2  job3      3

However, I'm getting:

keyerror:"['jo1'] not found in axis"

What am I doing wrong?

jobs_df = jobs_df[jobs_df['job_name'].isin(my_jobs)].

Quang Hoang
– Quang Hoang

2019-07-08 18:49:32 +00:00
Commented Jul 8, 2019 at 18:49 — Quang Hoang
– Quang Hoang, Commented Jul 8, 2019 at 18:49
@QuangHoang Thank you!!

Sarah
– Sarah

2019-07-08 19:13:08 +00:00
Commented Jul 8, 2019 at 19:13 — Sarah
– Sarah, Commented Jul 8, 2019 at 19:13

Sho arata · Accepted Answer · 2019-07-08 19:29:57Z

1

I believe the cause of the error is that it is looking for the job_name in the dataframe index which only has [0, 1, 2] you could set the column job_name as index:

jobs_df.set_index("job_name", inplace=True)

and then do:

my_jobs = ["job2", "job3"]
for job in jobs_df.index:
    if job not in my_jobs:
        jobs_df.drop(job, axis=0, inplace=True)
jobs_df.reset_index(inplace=True)

but an easier and faster way to do this would be to just do:

jobs_df = jobs_df[jobs_df["job_name"].isin(my_jobs)]

edited Jul 8, 2019 at 19:29

answered Jul 8, 2019 at 19:02

Sho arata

866 bronze badges

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

2 Comments

Sarah Over a year ago

Hi, is ~ necessary? The code worked, however I think it did not give me the correct result.. I assume ~ means No?

Akaisteph7 Over a year ago

@sarahkim You are right, the ~ means not. You do not need it. it is just a mistake.

Collectives™ on Stack Overflow

How to delete rows in a data frame with for loop

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