Python 提取字符串中的数字

在处理文本数据时，有时需要提取字符串中的数字。Python 提供了多种方法来实现这个功能。本文将详细介绍如何使用 Python 提取字符串中的数字，包括使用字符串方法、正则表达式以及第三方库等不同的方法。

方法一：使用字符串方法

Python 字符串类型提供了多个方法用于处理字符串。其中，isdigit() 方法可以判断一个字符串是否只包含数字字符。如果字符串只包含数字字符，则返回 True，否则返回 False。我们可以利用这个方法来提取字符串中的数字。

示例代码如下：

def extract_numbers_from_string(string):
    result = []
    current_number = ''

    for char in string:
        if char.isdigit():
            current_number += char
        elif current_number:
            result.append(int(current_number))
            current_number = ''

    if current_number:
        result.append(int(current_number))

    return result

string1 = 'abc123def456xyz789'
numbers1 = extract_numbers_from_string(string1)
print(numbers1)  # 输出：[123, 456, 789]

string2 = 'abcxyz'
numbers2 = extract_numbers_from_string(string2)
print(numbers2)  # 输出：[]

上述代码定义了一个名为 extract_numbers_from_string() 的函数，该函数接受一个字符串作为参数，并返回提取出的数字列表。在函数中，我们使用一个循环遍历字符串中的每个字符。如果字符是数字字符，则将其添加到 current_number 变量中；如果字符不是数字字符，且 current_number 不为空，则将其转换为整数，并添加到结果列表中。最后，如果 current_number 不为空，则将其转换为整数，并添加到结果列表的末尾。最终返回结果列表。

运行上述代码，可以看到输出结果为 [123, 456, 789]，符合预期。

方法二：使用正则表达式

正则表达式是用于匹配和处理字符串的强大工具。Python 的 re 模块提供了一系列函数来支持正则表达式操作。我们可以使用正则表达式来匹配和提取字符串中的数字。

示例代码如下：

import re

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

string1 = 'abc123def456xyz789'
numbers1 = extract_numbers_from_string(string1)
print(numbers1)  # 输出：[123, 456, 789]

string2 = 'abcxyz'
numbers2 = extract_numbers_from_string(string2)
print(numbers2)  # 输出：[]

上述代码使用了 re.findall() 函数来匹配所有满足指定正则表达式的子字符串，并返回一个列表。正则表达式 r'\d+' 可以匹配一个或多个数字字符。最后，我们将匹配到的数字字符串转换为整数，并返回结果列表。

运行上述代码，可以看到输出结果与方法一相同。

方法三：使用第三方库

除了上述方法，还可以使用一些第三方库来实现字符串中数字的提取，例如 numpy、pandas 等。这些库提供了更丰富的字符串处理功能，并且效率更高。

示例代码如下：

import numpy as np

def extract_numbers_from_string(string):
    return [int(char) for char in string if char.isdigit()]

string1 = 'abc123def456xyz789'
numbers1 = extract_numbers_from_string(string1)
print(numbers1)  # 输出：[1, 2, 3, 4, 5, 6, 7, 8, 9]

string2 = 'abcxyz'
numbers2 = extract_numbers_from_string(string2)
print(numbers2)  # 输出：[]

上述代码使用了列表推导式来提取字符串中的数字。如果字符是数字字符，则将其转换为整数，并添加到结果列表中。

运行上述代码，可以看到输出结果与前两种方法略有不同。这是因为使用了列表推导式来提取每个数字字符，而不是提取整个数字。