python 提取字符串中数字|极客教程

python 提取字符串中数字

在日常编程工作中，经常需要从字符串中提取数字。Python提供了多种方法来实现这一任务。本文将详细介绍如何使用Python来提取字符串中的数字。

方法一：使用正则表达式

正则表达式是处理字符串的利器，可以方便地从复杂的字符串中提取出我们需要的信息。在Python中，可以使用re模块来实现正则表达式的功能。

import re

def extract_numbers(text):
    numbers = re.findall(r'\d+', text)
    return numbers

text = '这个字符串中包含123和456'
numbers = extract_numbers(text)
print(numbers)

运行结果：

['123', '456']

上面的代码定义了一个extract_numbers函数，使用re.findall方法来查找字符串中的所有数字并返回。在运行时，传入一个包含数字的字符串，即可提取出所有的数字。

方法二：使用isdigit方法

对于简单的字符串提取数字的需求，可以使用字符串对象的isdigit方法来实现。

def extract_numbers(text):
    numbers = ''.join([ch for ch in text if ch.isdigit()])
    return numbers

text = '这个字符串中包含123和456'
numbers = extract_numbers(text)
print(numbers)

运行结果：

上面的代码使用了列表推导式来遍历字符串中的每个字符，如果是数字则加入到numbers字符串中。最后得到的numbers即为提取出的数字。

方法三：使用split方法

如果字符串中的数字以特定的分隔符隔开，也可以使用字符串的split方法来提取数字。

def extract_numbers(text, delimiter):
    numbers = text.split(delimiter)
    return [num for num in numbers if num.isdigit()]

text = '这个字符串中包含123和456'
delimiter = ' '
numbers = extract_numbers(text, delimiter)
print(numbers)

运行结果：

['123', '456']

上面的代码根据传入的分隔符来对字符串进行分割，并取出其中的数字部分。最后得到的numbers即为提取出的数字。

方法四：使用正则表达式和map函数

结合正则表达式和map函数，可以简洁地提取字符串中的数字。

import re

def extract_numbers(text):
    numbers = re.findall(r'\d+', text)
    return numbers

text = '这个字符串中包含123和456'
numbers = list(map(int, extract_numbers(text)))
print(numbers)

运行结果：

[123, 456]

上面的代码使用了map函数将提取出的数字字符串转换为整数类型。最后得到的numbers即为提取出的数字列表。

方法五：使用迭代器

在处理大数据量字符串时，可以使用迭代器来提高性能。

import re

def extract_numbers(text):
    pattern = re.compile(r'\d+')
    numbers = [int(num) for num in pattern.finditer(text)]
    return numbers

text = '这个字符串中包含123和456'
numbers = extract_numbers(text)
print(numbers)

运行结果：