pandas 筛选 isin|极客笔记

pandas 筛选 isin

在 pandas 中，我们经常需要根据一组特定的值来筛选数据集。而 isin 方法则是一种非常有效的筛选方式，它可以帮助我们快速地筛选出包含指定值的行或列。本文将详细介绍 isin 方法的用法及示例，帮助大家更好地理解和运用这一功能。

1. isin 方法简介

isin 方法是 pandas Series 和 DataFrame 对象自带的一个函数，用于判断每个元素是否包含在传入的列表或数组中。它返回一个布尔值的 Series，其中为 True 的元素表示对应位置的元素包含在传入的列表中，为 False 的元素表示对应位置的元素不包含在传入的列表中。

2. isin 方法的语法

isin 方法的语法如下：

DataFrame.isin(values)

其中，values 可以是列表、数组、Series 或者 DataFrame，用于指定要筛选的值。

3. isin 方法的示例

3.1 对 Series 使用 isin 方法

首先，我们创建一个简单的 Series：

import pandas as pd

data = {'A': [1, 2, 3, 4, 5],
        'B': ['foo', 'bar', 'baz', 'qux', 'quux']}

df = pd.DataFrame(data)
s = df['B']
print(s)

输出：

0     foo
1     bar
2     baz
3     qux
4    quux
Name: B, dtype: object

接下来，我们使用 isin 方法筛选出包含特定值的行：

result = s.isin(['foo', 'qux'])
print(result)

输出：

0     True
1    False
2    False
3     True
4    False
Name: B, dtype: bool

3.2 对 DataFrame 使用 isin 方法

接着，我们再创建一个较为复杂的 DataFrame：

data = {'A': [1, 2, 3, 4, 5],
        'B': ['foo', 'bar', 'baz', 'qux', 'quux'],
        'C': [0.1, 0.2, 0.3, 0.4, 0.5]}

df = pd.DataFrame(data)
print(df)

输出：

   A     B    C
0  1   foo  0.1
1  2   bar  0.2
2  3   baz  0.3
3  4   qux  0.4
4  5  quux  0.5

然后，我们使用 isin 方法筛选出包含特定值的行：

result = df[df['B'].isin(['foo', 'qux'])]
print(result)

输出：

   A    B    C
0  1  foo  0.1
3  4  qux  0.4

4. isin 方法的注意事项

在使用 isin 方法时，需要注意以下几点：

values 参数可以是列表、数组、Series 或者 DataFrame。
对于 DataFrame，可以指定要筛选的列名，也可以对整个 DataFrame 进行筛选。
isin 方法返回的是一个布尔值的 Series，可以直接作为索引使用或者与其他条件组合使用。

5. 总结

本文介绍了 pandas 中 isin 方法的基本用法和示例，希望能帮助大家更好地理解和运用这一功能。在实际数据处理中， isin 方法是一个非常实用的工具，能够帮助我们快速高效地筛选数据集，提高数据处理的效率。

pandas 筛选 isin