Python提取文件名数字的实用方法

本文详解了如何用Python高效提取文件名中的数字，核心方法是灵活运用正则表达式——通过`re.findall(r'\d+', filename)`一键捕获所有数字片段，并支持类型转换、批量处理目录文件以及精准定位开头或扩展名前的特定数字，配合`os.listdir()`或`pathlib`即可轻松应对实际项目中千变万化的文件命名规则，让数据预处理更智能、更可靠。

要从文件名中提取数字，可以使用正则表达式或字符串处理方法。最常用且高效的方式是使用 re 模块来匹配文件名中的数字部分。

使用正则表达式提取数字

导入 re 模块，用 re.findall() 找出所有数字片段：

import re
filename = "data_2023_file_45.txt"
numbers = re.findall(r'\d+', filename)
print(numbers)  # 输出: ['2023', '45']

\d+ 表示匹配一个或多个连续的数字。返回的是字符串列表，如需整数可转换：

numbers_int = [int(n) for n in numbers]
print(numbers_int)  # 输出: [2023, 45]

从多个文件中提取数字

结合 os.listdir() 或 pathlib 遍历目录：

import os
import re
folder = "./data/"
for file in os.listdir(folder):
match = re.findall(r'\d+', file)
if match:
print(f"{file} -> {match}")

只提取特定位置的数字

如果只想取文件名开头或结尾的数字，可用更精确的正则：

• 开头数字：^\\d+
• 结尾数字（在扩展名前）：\\d+(?=\\.txt) 或 \\d+(?=\\.)

# 提取以数字结尾的文件（不含扩展名）
filename = "image_123.jpg"
number_at_end = re.findall(r'(\d+)(?=\.\w+$)', filename)
print(number_at_end)  # 输出: ['123']

基本上就这些。根据命名规则选择合适的正则模式，就能准确读取文件名中的数字。

好了，本文到此结束，带大家了解了《Python提取文件名数字的实用方法》，希望本文对你有所帮助！关注golang学习网公众号，给大家分享更多文章知识！