pass both string and list to pandas .isin method

Question

I am trying to pass both a string and a list to the pandas .isin() method. Here is my code below

    overall_months = ['APR', 'JUL', 'NOV', 'MAR', 'FEB', 'AUG', 'SEP', 'OCT', 'JAN', 'DEC', 'MAY', 
    'JUN', ['APR', 'JUL', 'NOV', 'MAR', 'FEB', 'AUG', 'SEP', 'OCT', 'JAN', 'DEC', 'MAY', 'JUN']]

    for mon in overall_months:
        temp_df = df.month.isin([[mon]]))

The issue here is the .isin([]) is fine for each iteration of a string, but when i get to overall_months[-1], its a list and you cannot pass a list into .isin([]) syntax. Ive tried this but cannot remove the double quotes because my understanding is strings are immutable:

    str(overall_months[-1]).replace('[', '').replace(']','')

This produces: "'APR', 'JUL', 'NOV', 'MAR', 'FEB', 'AUG', 'SEP', 'OCT', 'JAN', 'DEC', 'MAY', 'JUN'" It could be passed to my syntax if it was: 'APR', 'JUL', 'NOV', 'MAR', 'FEB', 'AUG', 'SEP', 'OCT', 'JAN', 'DEC', 'MAY', 'JUN'

Any help in the best way to accomplish this?

Ian · Accepted Answer · 2024-06-29 21:23:18Z

1

You can check if the element is a list with isinstance:

for mon in overall_months:
    if not isinstance(mon, list): 
        mon = [mon]
    tmp_df = df.month.isin(mon)

edited Jun 29, 2024 at 21:23

Ian

164 bronze badges

answered Jul 29, 2020 at 20:02

Quang Hoang

151k11 gold badges64 silver badges86 bronze badges

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

Collectives™ on Stack Overflow

pass both string and list to pandas .isin method

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