問題描述
dataframe的某一列均為字符串格式,想篩選出含有特定字符串的行,具體實現代碼如下:
df[df['地址'].str.contains('北京')
這樣就可以篩選出‘地址’這一列中含有‘北京’所對應的行了。
問題升級
但如果需要篩選不止一個地名,比如包含‘北京’、‘天津’、‘上海’...等城市所對應的行,經過測試,直接放入列表不行!
解決方法是將關鍵詞都使用|
進行連接,這樣就可以一次性篩選出來了。
citys = ['北京', '天津', '上海']
address = '|'.join(citys)
df_new = df[df['地址'].str.contains(address)
這樣就一次性篩選出上述三個城市對應的行了。