Python从字符串只保留数字

原创

mob64ca12e60047 2024-09-24 06:32:43 ©著作权

文章标签 字符串 git Python 文章分类 Python 后端开发

©著作权归作者所有：来自51CTO博客作者mob64ca12e60047的原创作品，请联系作者获取转载授权，否则将追究法律责任

Python字符串处理：仅保留数字

在日常编程中，你可能会遇到需要从字符串中提取出数字的情况，例如清洗数据或进行格式化。Python提供了多种方法来处理字符串，这里我们将专注于提取字符串中的数字。通过本文，我们将介绍如何使用Python从字符串中只保留数字，并给出示例代码。

1. 简介

在现实生活中，数据可能以多种形式呈现，包括字母、符号和数字。当我们需要进行计算或者创建报表时，往往只需要提取出数字部分。Python的灵活性使得这一任务变得简单而直观。

2. 方法一：使用字符串的`join`和列表推导

一种常见的方法是使用字符串的join功能结合列表推导。我们可以遍历字符串中的每一个字符，检查它是否为数字，然后将这些数字连接成一个新的字符串。

以下是实现代码：

def filter_digits(input_string):
    return ''.join([char for char in input_string if char.isdigit()])

# 示例
input_str = "abc123!@#456def"
output_str = filter_digits(input_str)
print(output_str)  # 输出: 123456

在这个例子中，filter_digits函数接受一个字符串input_string，并返回一个只包含数字的新字符串。利用char.isdigit()方法，我们可以判断字符是否为数字。

3. 方法二：使用正则表达式

另一种更加灵活且强大的方法是使用Python的re模块。正则表达式允许我们使用模式匹配来提取字符串中的特定字符。

以下是使用正则表达式提取数字的示例代码：

import re

def filter_digits_regex(input_string):
    return ''.join(re.findall(r'\d', input_string))

# 示例
input_str = "abc123!@#456def"
output_str = filter_digits_regex(input_str)
print(output_str)  # 输出: 123456

在这个代码中，re.findall函数会查找字符串中所有与正则表达式\d匹配的部分，即数字字符。然后，通过join将它们组合成一个新的字符串。

4. 对比两种方法

sequenceDiagram
    participant User
    participant Method1 as 方法一: 字符串操作
    participant Method2 as 方法二: 正则表达式

    User->>Method1: 调用filter_digits
    Method1->>User: 返回只含数字的字符串
   
    User->>Method2: 调用filter_digits_regex
    Method2->>User: 返回只含数字的字符串

从上面的序列图中可以看出，用户可以选择任意一种方法来提取字符串中的数字。第一种方法更加直观，适合简单的操作；而第二种方法则更为灵活，适合复杂的模式匹配。