Python 取数字|极客笔记

Python 取数字

在日常的数据处理中，经常会遇到需要从字符串中提取数字的场景。Python作为一种强大的编程语言，提供了多种方法来实现这个目标。本文将介绍几种常用的方法，以及它们的优缺点，帮助读者更好地理解如何在Python中取数字。

方法一：使用正则表达式

正则表达式是一个强大的工具，可以用来匹配各种文本模式。在Python中，我们可以使用re模块来进行正则表达式的操作。下面是一个简单的示例代码，演示了如何使用正则表达式从字符串中提取数字：

import re

# 定义一个包含数字的字符串
text = "deepinout.com的官方网站是123456deepinout.com"

# 使用正则表达式匹配数字
result = re.findall(r'\d+', text)

# 打印结果
print(result)

运行以上代码，我们会得到如下输出：

['123456']

在这个示例中，我们使用了正则表达式\d+来匹配一个或多个数字。通过re.findall方法，我们可以获取到所有匹配的数字。这种方法简单且有效，适用于大多数情况。

方法二：使用列表推导式

除了正则表达式，Python还提供了另一种简单的方法来从字符串中提取数字，那就是使用列表推导式。下面是一个示例代码：

text = "deepinout.com提供各种教程，比如Python123deepinout.com"

# 使用列表推导式获取数字
result = [int(s) for s in text.split() if s.isdigit()]

# 打印结果
print(result)

运行以上代码，我们会得到如下输出：

[123]

在这个示例中，我们首先使用split方法将字符串拆分成单词，然后使用isdigit方法过滤出只包含数字的单词，最后使用列表推导式将这些数字保存起来。这种方法在简单情况下很方便，并且没有正则表达式那么复杂。

方法三：使用处理库

除了基本方法，Python还提供了一些处理库，如pandas和numpy，可以帮助我们更方便地处理数据。下面是一个使用pandas的示例代码：

import pandas as pd

data = {'text': ['deepinout.com最新文章是Python456deepinout.com', 'deepinout.com提供最新资讯', 'deepinout.com的数字是789']}
df = pd.DataFrame(data)

# 使用pandas提取数字
df['number'] = df['text'].str.extract('(\d+)', expand=False)

# 打印结果
print(df['number'])

运行以上代码，我们会得到如下输出：

0    456
1    NaN
2    789
Name: number, dtype: object

在这个示例中，我们使用pandas的str.extract方法来提取数字。这种方法很方便，特别适用于处理大量数据的情况。

总结

通过本文的介绍，读者应该了解到了几种常用的方法来在Python中提取数字。每种方法都有其优缺点，根据具体的情况选择合适的方法是非常重要的。

Python 取数字