Python抽取字符串中的数字|极客笔记

Python抽取字符串中的数字

在处理文本数据时，经常会遇到需要从字符串中抽取数字的情况。在Python中，有多种方法可以实现这一需求。本文将详细介绍如何使用正则表达式、内置函数及第三方库来实现从字符串中抽取数字的操作。

正则表达式

正则表达式是一种强大的字符串匹配工具，可以用来快速查找、替换或抽取字符串中的特定模式。我们可以使用正则表达式来匹配并提取字符串中的数字。

import re

def extract_numbers(text):
    numbers = re.findall(r'\d+', text)
    return [int(number) for number in numbers]

text = "There are 123 apples, 456 pears, and 789 bananas."
numbers = extract_numbers(text)
print(numbers)

运行结果：

[123, 456, 789]

在上面的示例代码中，我们使用re.findall函数来查找文本中的所有数字模式（\d+表示匹配一个或多个数字字符）。然后通过列表推导式将结果转换为整数列表。

内置函数

除了正则表达式外，Python的内置函数也提供了一些用于抽取字符串中的数字的方法。其中isdigit()和isnumeric()函数可以用来判断一个字符串是否只包含数字字符。

def extract_numbers(text):
    numbers = [int(char) for char in text if char.isdigit()]
    return numbers

text = "There are 123 apples, 456 pears, and 789 bananas."
numbers = extract_numbers(text)
print(numbers)

运行结果：

[1, 2, 3, 4, 5, 6, 7, 8, 9]

在上面的示例代码中，我们使用列表推导式和isdigit()函数来过滤文本中的数字字符，并将其转换为整数列表。

第三方库

除了使用内置函数和正则表达式外，还可以使用第三方库来辅助抽取字符串中的数字。其中NumPy和Pandas是常用的数据处理库，它们提供了许多方法来处理数字数据。

import numpy as np
import pandas as pd

text = "There are 123 apples, 456 pears, and 789 bananas."
numbers = [int(char) for char in text if char.isdigit()]
series = pd.Series(numbers)
mean = np.mean(series)

print(series)
print("Mean:", mean)

运行结果：

0    1
1    2
2    3
3    4
4    5
5    6
6    7
7    8
8    9
dtype: int64
Mean: 5.0

在上面的示例代码中，我们将抽取到的数字字符转换为整数列表，并使用Pandas库创建了一个Series对象。最后使用NumPy库来计算数字的均值。

总结而言，本文介绍了在Python中抽取字符串中的数字的几种方法，包括使用正则表达式、内置函数和第三方库。开发人员可以根据具体需求选择适合的方法来处理文本数据中的数字信息。

Python抽取字符串中的数字