パンダ-isin

#!/usr/bin/env python
# coding: utf-8

# # 第一课 数据分析工具Pandas高阶
# ## 第四节 isin函数

# In[1]:


import pandas as pd
import numpy as np
# * Series.isin()
# In[2]:
s = pd.Series(np.arange(5), 
              index=['a', 'b', 'c', 'd', 'e'])
# In[3]:
s
# In[4]:
a    0
b    1
c    2
d    3
e    4
dtype: int32

# 返回值为布尔值向量
s.isin([2, 4, 6])
# In[5]:
a    False
b    False
c     True
d    False
e     True
dtype: bool
# 数据过滤
s[s.isin([2, 4, 6])]
# * Index.isin()
# In[6]:
s.index.isin(['c', 'd'])
# In[7]:
s[s.index.isin(['c', 'd'])]
# * DataFrame.isin()
# In[8]:
df = pd.DataFrame({
    
    'col1': [1, 2, 3, 4], 
                   'col2': ['a', 'b', 'f', 'n'],
                   'col3': ['a', 'n', 'c', 'n']})
# In[9]:
df
# In[10]:
# 所有值过滤
vals1 = ['a', 'b', 1, 3]
df.isin(vals1)
# In[11]:
df[df.isin(vals1)]
# In[12]:
# 按列过滤
vals2 = {
    
    'col1': [1, 3],
         'col2': ['a', 'b']}
df.isin(vals2)
# In[13]:
df[df.isin(vals2)]
# In[ ]: