pandas 實現模糊檢索

 

1  生成一個dataframe

>>> import pandas as pd                

  >>> df = pd.read_excel('students.xlsx', index_col='ID')
  >>> df
   name class
ID
1 張三 一班
2 張三 二班
3 李四 二班
4 王五 一班
5 麻六 三班
6 喬七 三班

2.1 函數loc定位cell

定位一個元素df.loc[‘行名’, '列名']

>>> df.loc[3,'name']
'李四'

# 注意：3是行名，而不是行號

2.2 函數iloc定位cell

df.iloc[行號, 列號]

>>> df.iloc[2, 0]
'李四'

3 通過.str.contains()實現模糊檢索

比如，查找二班的張三

>>> df.loc[df['name'].str.contains('張') & df['class'].str.contains('二')]   
   name class                                                              
ID                                                                         
2    張三    二班                                                              

注意：如果中間的.str不用的話，就會出錯，提示‘Series’數組沒有‘contains'函數。