pandas 选择列值为数字的内容
时间：2023-11-08 11:11:18 标签：pandas
python
数字
看过来
《pandas 教程》 持续更新中，可作为 pandas 入门进阶课程、pandas 中文手册、用法大全，配有案例讲解和速查手册。提供建议、纠错、催更等加作者微信: gairuo123（备注：pandas教程）和关注公众号「盖若」ID: gairuo。查看更新日志。作者开办 Python 数据分析训练营正在报名中，详情 Python 数据分析训练营。
|
本教程作者所著新书《Python之光：Python编程入门与实战》（ISBN：9787111729891）已由机械工业出版社出版上市，各大电商平台有售，欢迎：查看详情并关注购买。 |
|
本教程作者所著新书《深入浅出Pandas：利用Python进行数据处理与分析》（ISBN：9787111685456）已由机械工业出版社出版上市，各大电商平台有售，欢迎：查看详情并关注购买。 |
我们有一个 DataFrame，其中有一个列需要我们重点关注，如它的数据为非数字类型，则删除这行的数据。我们来看看 pandas 如何完成。
数据与需求
我们的数据构造如下：
import pandas as pd
import io
data = '''
EID Name
123 Madsen,Gunnar
ret Greene,Richard
465 Stull,Matthew
'''
df = pd.read_csv(io.StringIO(data), delim_whitespace=True)
df
'''
EID Name
0 123 Madsen,Gunnar
1 ret Greene,Richard
2 465 Stull,Matthew
'''
注意 EID 列，索引 1 行的值为非数字，我们需要将这部分剔除，保留其它为数字的数据行。
思路
我们可以用字符串的 str.isdecimal 方法来判断这个字符串是否一个数字，得到一个布尔序列，然后再用这个布尔序列来抽取整体 DataFrame 数据。
代码
判断数据是否数字：
df.EID.apply(str.isdecimal)
'''
0 True
1 False
2 True
Name: EID, dtype: bool
'''
将布尔序列来抽取整体 DataFrame 数据：
df[df.EID.apply(str.isdecimal)]
'''
EID Name
0 123 Madsen,Gunnar
2 465 Stull,Matthew
'''
其他方法
以下是分别用 pd.to_numeric() 函数和正则表达式的两种方法。
df[pd.to_numeric(df.EID, errors='coerce').notna()]
df[df.EID.str.match(r'\d+')] # 推荐！
这样就完成了需求。
（完）
相关内容
- pandas 实战案例集 2023-09-01 11:21:28
- pandas 查询筛选数据 2023-10-25 21:32:42
- Python str.isdecimal() 是否全为十进制数字 2023-11-29 09:59:20
