python pandas.Series.isin 不区分大小写

Question

我想用数据在列表中的 DataFrame 的列之一过滤掉一些行。

df[df['column'].isin(mylist)]

但是我发现它是区分大小写的。有没有使用不区分大小写的“.isin()”的方法？

Answer 1

一种方法是比较系列的小写或大写与列表的相同

df[df['column'].str.lower().isin([x.lower() for x in mylist])]

这里的优点是我们不保存对原始 df 或列表的任何更改，从而使操作更加高效

考虑这个虚拟 df:

    Color   Val
0   Green   1
1   Green   1
2   Red     2
3   Red     2
4   Blue    3
5   Blue    3

对于列表l:

l = ['green', 'BLUE']

你可以使用 isin()

df[df['Color'].str.lower().isin([x.lower() for x in l])]

你得到

    Color   Val
0   Green   1
1   Green   1
4   Blue    3
5   Blue    3