python中isin函数的用法

isin函数用于筛选pandas DataFrame或Series中存在于给定列表或数组中的元素。

在Python编程中，isin函数是一个常见的用于数据过滤的函数，它主要用于Pandas库中的DataFrame或Series对象。isin函数的主要功能是返回一个布尔型的Series或DataFrame，表示每个元素是否包含在给定的值集合中。

isin函数的基本语法如下：

DataFrame.isin(values)

values参数可以是一个列表、集合、字典、Series或DataFrame，如果values是字典，那么其键值对将被用来确定对应列是否存在于DataFrame中。

假设我们有一个DataFrame如下：

import pandas as pd
data = {'A': [1, 2, 3], 'B': [4, 5, 6]}
df = pd.DataFrame(data)

我们可以使用isin函数来找出’A’列中值为1或2的所有行：

print(df['A'].isin([1, 2]))

输出结果为：

0     True
1     True
2    False
Name: A, dtype: bool

这表示第0行和第1行的’A’列的值都在给定的列表[1, 2]中。

除了基本用法，isin函数还有一些高级用法，我们可以使用~操作符来获取isin的反向结果，即找出不在给定值集合中的元素。

print(~df['A'].isin([1, 2]))

输出结果为：

0    False
1    False
2     True
Name: A, dtype: bool

这表示第0行和第1行的’A’列的值都不在给定的列表[1, 2]中，而第2行的’A’列的值在给定的列表[1, 2]中。

我们还可以同时对多个列进行isin操作，我们可以找出’A’列值为1或2，且’B’列值为4或5的所有行：

print(df[(df['A'].isin([1, 2])) & (df['B'].isin([4, 5]))])

输出结果为：

   A  B
0  1  4
1  2  5

这表示第0行和第1行满足条件。