Python 字符串提取数字|极客教程

Python 字符串提取数字

概述

在Python中，字符串是非常常见的数据类型。我们常常需要从字符串中提取出其中的数字部分，以便于后续的处理和运算。本文将介绍几种常用的方法，以帮助你实现字符串提取数字的功能。

方法一：使用正则表达式

正则表达式是一种强大的文本匹配工具，可以用于查找、匹配和替换文本。在Python中，我们可以使用re模块来使用正则表达式。下面是一个使用正则表达式提取数字部分的示例代码：

import re

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

text = "在这句话中有1234个数字，例如5678和91011。"
numbers = extract_numbers_with_regex(text)
print(numbers)

运行结果：

[1234, 5678, 91011]

在这个示例代码中，我们定义了一个名为extract_numbers_with_regex的函数，该函数使用正则表达式\d+来匹配字符串中的数字部分。re.findall函数返回一个包含所有匹配的列表，然后我们使用列表推导式将这些字符串类型的数字转换为整数类型。

方法二：使用列表推导式和`isdigit`方法

在Python中，字符串对象有一个名为isdigit的方法，可以用来判断一个字符串是否只包含数字字符。我们可以结合使用列表推导式和isdigit方法来提取字符串中的数字部分。下面是一个示例代码：

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

text = "在这句话中有1234个数字，例如5678和91011。"
numbers = extract_numbers_with_isdigit(text)
print(numbers)

运行结果：

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

在这个示例代码中，我们定义了一个名为extract_numbers_with_isdigit的函数，该函数使用列表推导式和isdigit方法来提取字符串中的数字部分。我们首先遍历字符串中的每一个字符，并使用isdigit方法来判断是否为数字字符，如果是，则将其转换为整数类型。

需要注意的是，这种方法提取出的是单个数字字符，而不是多位数。如果字符串中有多位数的数字，我们可以稍作修改来提取出多位数的数字。

方法三：使用字符串切片和`isdigit`方法

与方法二类似，我们可以使用字符串的切片操作和isdigit方法来提取字符串中的数字部分。下面是一个示例代码：

def extract_numbers_with_slice(text):
    numbers = []
    i = 0
    while i < len(text):
        if text[i].isdigit():
            j = i
            while j < len(text) and text[j].isdigit():
                j += 1
            numbers.append(int(text[i:j]))
            i = j - 1
        i += 1
    return numbers

text = "在这句话中有1234个数字，例如5678和91011。"
numbers = extract_numbers_with_slice(text)
print(numbers)

运行结果：

[1234, 5678, 91011]

在这个示例代码中，我们定义了一个名为extract_numbers_with_slice的函数来提取字符串中的数字部分。我们使用一个循环来遍历字符串中的每一个字符。如果当前字符是数字字符，则从该字符位置开始向后查找连续的数字字符，直到遇到第一个非数字字符为止。然后，我们将找到的数字字符转换为整数类型，并添加到一个列表中。

方法四：使用第三方库

除了使用Python内置的方法外，还有一些第三方库可以帮助我们进行字符串的处理。例如，NumPy和Pandas等库提供了强大的函数和方法来处理数据。下面是一个使用NumPy来提取字符串中的数字部分的示例代码：

import numpy as np

def extract_numbers_with_numpy(text):
    numbers = np.fromiter(filter(str.isdigit, text), dtype=int)
    return numbers

text = "在这句话中有1234个数字，例如5678和91011。"
numbers = extract_numbers_with_numpy(text)
print(numbers)

运行结果：