python中isin用法

Python中的isin()方法用于筛选出列表中存在于指定集合中的元素。
创新互联公司始终坚持【策划先行，效果至上】的经营理念，通过多达十余年累计超上千家客户的网站建设总结了一套系统有效的全网整合营销推广解决方案，现已广泛运用于各行各业的客户，其中包括：成都三维植被网等企业，备受客户称扬。

在Python的Pandas库中，isin()是一个非常实用的函数，主要用于筛选数据，这个函数可以检查DataFrame或Series对象中的值是否包含在给定的值序列中。

基本用法

isin()函数的基本用法非常简单，它接收一个列表作为参数，然后返回一个布尔型的DataFrame或Series，其中的每个元素都表示原始DataFrame或Series中的相应元素是否在给定的列表中。

假设我们有一个包含多个城市名称的DataFrame：

import pandas as pd
data = {'City': ['New York', 'Los Angeles', 'Chicago', 'Houston', 'Phoenix']}
df = pd.DataFrame(data)

如果我们想要找出在这个DataFrame中哪些城市是在美国的前五大城市（按照人口排序），我们可以使用isin()函数：

top_five_cities = ['New York', 'Los Angeles', 'Chicago', 'Houston', 'Phoenix']
result = df['City'].isin(top_five_cities)

在这个例子中，result将是一个布尔型的Series，表示每个城市是否在前五大城市中。

高级用法

除了基本用法之外，isin()函数还有一些高级用法。

1. 使用字典进行筛选

如果你有一个字典，其中键是列名，值是要检查的值的列表，你也可以使用isin()函数。

data = {'City': ['New York', 'Los Angeles', 'Chicago', 'Houston', 'Phoenix'],
        'State': ['New York', 'California', 'Illinois', 'Texas', 'Arizona']}
df = pd.DataFrame(data)
criteria = {'City': ['New York', 'Los Angeles'], 'State': ['Texas']}
result = df.isin(criteria)

在这个例子中，result将是一个布尔型的DataFrame，表示每行中的每个元素是否在相应的列表中。

2. 使用正则表达式进行筛选

isin()函数还支持使用正则表达式进行筛选，你只需要将正则表达式作为字符串传递给isin()函数即可。

如果我们想要找出所有以"N"开头的城市，我们可以这样做：

pattern = '^N'
result = df['City'].str.contains(pattern)

在这个例子中，result将是一个布尔型的Series，表示每个城市的名称是否匹配给定的正则表达式。