Select columns from pandas dataframe using multiple conditions on columns in Python

Question

I have following pandas dataframe (umls)

             CUI      SDUI  SAB  TTY                    STR
325040  C0011405   D003788  MSH   MH   Dental Pulp Diseases
325054  C0011405  10012328  MDR  LLT   Dental pulp disorder
325055  C0011405  10012328  MDR   PT   Dental pulp disorder
325057  C0011405  10044050  MDR   HT  Dental pulp disorders
325061  C0011405   D003788  MSH  DEV          PULP DIS DENT
325062  C0011405   D003788  MSH  DEV          DENT PULP DIS
325063  C0011405   D003788  MSH  DEV          DIS DENT PULP

I would like to filter rows based on certain conditions like: When SAB = MSH, select TTY= MH and when SAB = MDR, select TTY= LLT and PT.

I am expecting below output:

             CUI      SDUI  SAB  TTY                    STR
325040  C0011405   D003788  MSH   MH   Dental Pulp Diseases
325054  C0011405  10012328  MDR  LLT   Dental pulp disorder
325055  C0011405  10012328  MDR   PT   Dental pulp disorder

I am using following lines of code:

umls[(umls['SAB'].isin(['MSH', 'MDR']))] & (umls['TTY'].isin(['MH', 'LLT', 'PT']))]

Any help is highly appreciated

the filtering may be simpler via a MultiIndex

sammywemmy
– sammywemmy

2022-10-13 12:14:48 +00:00
Commented Oct 13, 2022 at 12:14 — sammywemmy
– sammywemmy, Commented Oct 13, 2022 at 12:14

jezrael · Accepted Answer · 2022-10-13 11:48:04Z

2

Remove ()[] for chain both masks:

df = umls[umls['SAB'].isin(['MSH', 'MDR']) & umls['TTY'].isin(['MH', 'LLT', 'PT'])]
print (df)
             CUI      SDUI  SAB  TTY                   STR
325040  C0011405   D003788  MSH   MH  Dental Pulp Diseases
325054  C0011405  10012328  MDR  LLT  Dental pulp disorder
325055  C0011405  10012328  MDR   PT  Dental pulp disorder

answered Oct 13, 2022 at 11:48

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

Select columns from pandas dataframe using multiple conditions on columns in Python

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