python 给txt文件统一改名，保留一段字符

文件名的匹配：（我需要给所有txt文件改名）

1、假设在 R、N、W、Z 目录下，每个参数的 .txt 文件名是相同的。例如，R 目录下有一个名为 event1.txt 的文件，那么 N、W、Z 目录下也应该有一个名为 event1.txt 的文件。

2、这种假设是为了确保在加载数据时，可以正确地从一个文件名读取四个参数的数据。

文件夹很多的名字都是这种命名方法：

normalized_filtered_54031-20241125225400-20241126021159-0_N_clusum_one_001.txt  想变为 54031-20241125225400-20241126021159-0_001

import os

def rename_files(directory):
    # 遍历目录中的所有文件
    for filename in os.listdir(directory):
        if filename.endswith('.txt'):
            # 提取文件名中我们感兴趣的部分
            # 假设格式为 normalized_filtered_54031-20241125225400-20241126021159-0_N_clusum_one_001.txt
            # 我们要提取的部分是 "54031-20241125225400-20241126021159"
            parts = filename.split('_')
            if len(parts) >= 4:  # 确保有足够的部分来提取
                extracted_part = parts[2]  # 提取 "54031-20241125225400-20241126021159"
                number_part = parts[-1].split('.')[0]  # 提取 "001"
                # 构建新的文件名
                new_filename = f"{extracted_part}_{number_part}.txt"
                # 构造完整的旧文件路径和新文件路径
                old_path = os.path.join(directory, filename)
                new_path = os.path.join(directory, new_filename)
                # 重命名文件
                os.rename(old_path, new_path)
                # print(f"Renamed {filename} to {new_filename}")

# 使用示例
directory = '/training_data/Z/'  # 替换为你的目录路径
rename_files(directory)