pandas isin筛选

pandas中的isin()方法可以用来筛选某个DataFrame或Series中的数据是否在另一个DataFrame或Series中。具体使用方法如下：

1. 对于DataFrame，isin()方法可以使用一个Series、DataFrame、数组、列表等作为参数，用来筛选特定列中的数据是否在参数中存在。例如：

import pandas as pd

df = pd.DataFrame({'A': [1, 2, 3, 4], 'B': ['a', 'b', 'c', 'd']})
print(df)

#    A  B
# 0  1  a
# 1  2  b
# 2  3  c
# 3  4  d

# 筛选A列中数据是否在[2, 4, 6]中存在
mask = df['A'].isin([2, 4, 6])
print(mask)

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

# 根据mask筛选DataFrame
result = df[mask]
print(result)

#    A  B
# 1  2  b
# 3  4  d

2. 对于Series，isin()方法可以使用一个数组、列表等作为参数，用来筛选该Series中的数据是否在参数中存在。例如：

import pandas as pd

s = pd.Series(['a', 'b', 'c', 'd'])
print(s)

# 0    a
# 1    b
# 2    c
# 3    d
# dtype: object

# 筛选s中的数据是否在['c', 'd', 'e']中存在
mask = s.isin(['c', 'd', 'e'])
print(mask)

# 0    False
# 1    False
# 2     True
# 3     True
# dtype: bool

# 根据mask筛选Series
result = s[mask]
print(result)

# 2    c
# 3    d
# dtype: object

以上就是pandas中isin()方法的用法，希望对你有所帮助。