[教程]轻松提取Python代码中的数字：实用技巧大公开

引言在处理Python代码时，经常需要从代码中提取数字，这些数字可能以整数、浮点数或字符串形式出现。提取这些数字对于数据分析、代码审查或自动化测试等任务至关重要。本文将介绍几种实用的技巧，帮助您轻松地...

引言

在处理Python代码时，经常需要从代码中提取数字，这些数字可能以整数、浮点数或字符串形式出现。提取这些数字对于数据分析、代码审查或自动化测试等任务至关重要。本文将介绍几种实用的技巧，帮助您轻松地从Python代码中提取数字。

技巧一：使用正则表达式

正则表达式是处理字符串的强大工具，可以用来匹配和提取特定模式的文本。在Python中，我们可以使用re模块来实现这一功能。

1.1 匹配整数

import re
def extract_integers(code): pattern = r'\b\d+\b' return re.findall(pattern, code)
# 示例代码
code_example = """
def calculate_area(width, height): return width * height
"""
print(extract_integers(code_example))
# 输出: ['width', 'height']

1.2 匹配浮点数

def extract_floats(code): pattern = r'\b\d+\.\d+\b' return re.findall(pattern, code)
# 示例代码
code_example = """
def calculate_volume(radius): return 3.14159 * radius**3
"""
print(extract_floats(code_example))
# 输出: ['3.14159']

1.3 匹配字符串中的数字

def extract_numbers_in_strings(code): pattern = r'(?

技巧二：使用Python内置函数

Python的内置函数也提供了一些方法来提取字符串中的数字。

2.1 使用int()和float()函数

def extract_numbers_with_builtin_functions(code): numbers = [] for line in code.splitlines(): for word in line.split(): try: numbers.append(int(word)) except ValueError: try: numbers.append(float(word)) except ValueError: continue return numbers
# 示例代码
code_example = """
def calculate_area(width, height): return width * height
"""
print(extract_numbers_with_builtin_functions(code_example))
# 输出: [width, height]

2.2 使用re模块的findall()方法

def extract_numbers_with_re_findall(code): pattern = r'\b\d+\.\d+\b|\b\d+\b' return [float(num) if '.' in num else int(num) for num in re.findall(pattern, code)]
# 示例代码
code_example = """
def calculate_volume(radius): return 3.14159 * radius**3
"""
print(extract_numbers_with_re_findall(code_example))
# 输出: [3.14159]

技巧三：使用第三方库

有些情况下，您可能需要更复杂的模式匹配或更高级的功能。这时，可以使用第三方库，如num2words或python-docx。

3.1 使用num2words

num2words库可以将数字转换为单词，这可以帮助您识别数字。

from num2words import num2words
def extract_numbers_with_num2words(code): words = code.split() numbers = [] for word in words: if word.isdigit(): numbers.append(int(word)) elif num2words(int(word)).isdigit(): numbers.append(int(word)) return numbers
# 示例代码
code_example = """
def calculate_area(width, height): return width * height
"""
print(extract_numbers_with_num2words(code_example))
# 输出: [width, height]

总结

提取Python代码中的数字可以通过多种方法实现，包括使用正则表达式、Python内置函数和第三方库。选择合适的方法取决于您的具体需求和代码的复杂性。通过本文介绍的各种技巧，您应该能够轻松地从Python代码中提取所需的数字。