【python基础实例】从文件命名中提取特定信息（数字、字母等）

【python基础实例】从文件命名中提取特定信息（数字、字母等）

在处理文件和数据时，我们经常需要从文件名中提取特定的信息，如日期、序号或项目标识符。Python提供了强大的字符串处理功能，可以轻松完成这些任务。本文将通过一个基础实例，展示如何使用Python从复杂的文件命名中提取数字、字母等信息。
示例代码：

import os

# 定义一个函数来处理文件名并提取信息
def extract_info_from_filename(filename):
    # 使用os.path.basename获取纯文件名
    base_name = os.path.basename(filename)
    
    # 分割文件名以提取信息
    parts = base_name.split('_')
    
    # 初始化一个字典来存储提取的信息
    info = {
        'project': '',
        'year': '',
        'month': '',
        'type': '',
        'number': '',
        'theme': ''
    }
    
    # 假设文件名遵循特定的命名规则，例如 "projectname_year_month_type_number_theme.txt"
    if len(parts) >= 6:
        info['project'] = parts[0]
        info['year'] = parts[1]
        info['month'] = parts[2]
        info['type'] = parts[3]
        info['number'] = parts[4]
        info['theme'] = parts[5].split('.')[0]  # 去除文件扩展名
    
    return info

# 示例文件路径
file_path = '/path/to/your/data/projectA_2024_01_report_001_analysis.txt'

# 调用函数并打印结果
extracted_info = extract_info_from_filename(file_path)
print(extracted_info)

输出结果：

{
    'project': 'projectA',
    'year': '2024',
    'month': '01',
    'type': 'report',
    'number': '001',
    'theme': 'analysis'
}

在上述代码中，我们首先定义了一个函数extract_info_from_filename，它接受一个文件路径作为参数。使用os.path.basename函数从路径中提取文件名，然后使用split方法按照下划线’_'分割文件名。我们假设文件名遵循一定的命名规则，例如 “projectname_year_month_type_number_theme.txt”。通过分割得到的列表，我们可以提取出项目名称、年份、月份、报告类型、序号和主题，并将这些信息存储在一个字典中返回。

这个基础实例展示了如何使用Python的字符串处理功能来提取文件名中的关键信息。你可以根据自己的需求调整代码，以适应不同的文件命名规则和信息提取需求。