问题描述
dataframe的某一列均为字符串格式,想筛选出含有特定字符串的行,具体实现代码如下:
df[df['地址'].str.contains('北京')
这样就可以筛选出‘地址’这一列中含有‘北京’所对应的行了。
问题升级
但如果需要筛选不止一个地名,比如包含‘北京’、‘天津’、‘上海’...等城市所对应的行,经过测试,直接放入列表不行!
解决方法是将关键词都使用|
进行连接,这样就可以一次性筛选出来了。
citys = ['北京', '天津', '上海']
address = '|'.join(citys)
df_new = df[df['地址'].str.contains(address)
这样就一次性筛选出上述三个城市对应的行了。