pandas isin

在Pandas中，`isin`函数用于检查DataFrame或Series中的元素是否属于给定的列表、数组或Series。

下面是使用`isin`函数的示例：

import pandas as pd

# 创建一个示例DataFrame
data = {'Name': ['Alice', 'Bob', 'Charlie', 'David'],
        'Age': [25, 30, 35, 40],
        'City': ['New York', 'Paris', 'London', 'Tokyo']}
df = pd.DataFrame(data)

# 使用isin函数筛选出特定城市的数据
cities = ['Paris', 'London']
subset = df[df['City'].isin(cities)]
print(subset)

上述代码中，`df['City'].isin(cities)`返回一个布尔数组，其中元素为True表示对应行的'City'列值在给定的城市列表中。然后，将该布尔数组作为索引，筛选出满足条件的数据。输出结果如下：

      Name  Age    City
1      Bob   30   Paris
2  Charlie   35  London

`isin`函数还可以用于筛选Series中的数据。例如：

s = pd.Series([1, 2, 3, 4])
subset = s[s.isin([2, 4])]
print(subset)

上述代码中，`s.isin([2, 4])`返回一个布尔数组，其中元素为True表示对应位置的值在给定的列表中。然后，将该布尔数组作为索引，筛选出满足条件的数据。输出结果如下：

1    2
3    4
dtype: int64

通过使用`isin`函数，你可以方便地筛选出属于特定列表、数组或Series中的元素。