筛选包含某字符串的数据
import pandas as pd
data = pd.read_csv(path)
data
ts_code symbol name area industry list_date
0 000001.SZ 1 平安银行 深圳 银行 19910403
1 000002.SZ 2 万科A 深圳 全国地产 19910129
2 000004.SZ 4 ST国华 深圳 软件服务 19910114
3 000005.SZ 5 ST星源 深圳 环境保护 19901210
# 需求:筛选出即将退市(股票名称以ST开头)的股票信息
data["name"].str.contains("ST") #判断
0 False
1 False
2 True
3 True
Name: name, dtype: bool
data[data["name"].str.contains("ST")] #筛选
ts_code symbol name area industry list_date
2 000004.SZ 4 ST国华 深圳 软件服务 19910114
3 000005.SZ 5 ST星源 深圳 环境保护 19901210
# 需求:过滤出即将退市的股票
data[~data["name"].str.contains("ST")]
ts_code symbol name area industry list_date
0 000001.SZ 1 平安银行 深圳 银行 19910403
1 000002.SZ 2 万科A 深圳 全国地产 19910129