将string和list同时传递给pandas.isin方法

Question

我试图将一个字符串和一个列表传递给pandas.isin（）方法。下面是我的代码

    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]]))

这里的问题是，.isin([])对于字符串的每次迭代都很好，但是当我到达overall_months[-1]时，它是一个列表，您不能将列表传递到.isin([])语法中。我试过了，但不能删除双引号，因为我的理解是字符串是不可变的：

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

有什么最好的帮助来完成这件事吗？

Answer 1

您可以检查元素是否是带有isinstance的列表：

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