Python Pandas 筛选包含某字符串的行数据过滤含某字符串的数据

最新推荐文章于 2024-09-02 09:47:51 发布

原创最新推荐文章于 2024-09-02 09:47:51 发布 · 1.9k 阅读

14 ·

本内容遵循CC 4.0 BY-SA版权协议

标签

#1024程序员节 #pandas #python

收录于

python 专栏收录该内容

11 篇文章

订阅专栏

该篇博客展示了如何利用Pandas库在Python中筛选含有特定字符串的数据行，例如找出股票名称以'ST'开头，代表即将退市的股票。通过`data[name].str.contains(ST)`进行判断，然后使用条件索引实现筛选与过滤操作。

筛选包含某字符串的数据

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