How to exclude some rows from a dataframe

Question

I need help to exclude some rows: I would like for example to exclude three patients from my analysis (patients 1, 2, 6) Can you tell me why this code doesn't work and what I should write?

baseline_all_patients=baseline[baseline["studyid"]!=(1, 2, 6)]

score 0 · Accepted Answer · edited May 06 '20 at 15:21

0

You can try the following:

mask_patient = (~baseline["studyid"].isin([1, 2, 6]))
baseline_all_patients = baseline[mask_patient]

edited May 06 '20 at 15:21

UJIN

1,648
13
28

answered May 06 '20 at 15:15

Kosmos

497
3
11

score 0 · Answer 2 · answered May 06 '20 at 15:17

I see 2 issues with the thing you tried:

lists use square brackets
you want the studyids not IN the list, rather than equal to the list.

I think this does what you're trying to do:

import pandas as pd
baseline = pd.DataFrame({'studyid':range(10),'info':range(100,90,-1)})
baseline[~baseline.studyid.isin([1,2,6])]

How to exclude some rows from a dataframe

2 Answers2