Python文件名分割与列表匹配问题求助：如何实现预期分割效果？

问题描述 #

我正在做一个字符串与列表部分匹配的项目，需求是用Python遍历条件列表，检查指定目录下的文件名是否包含列表中的匹配项。

我写了一段代码，但运行结果不符合预期：

import pandas as pd
import os
import re

from re import split

df = pd.read_excel(r"C:\Users\Asus\Desktop\py\con.xlsx")
id = df.iloc[:,3].tolist()
file_dir = "C:\Users\Asus\Desktop\Python\Input"

#read filenames and split filename
for x in os.listdir(file_dir):
    f_name = os.path.basename(x[:-4])
    print(f_name)
    for fname in f_name:
        new_fn = fname.split(r'_|-', 6) #here I wanted to set max 6 split
        print(new_fn)

#set conditions loop, rename file
if any(ele in new_fn for ele in id):
  print("Success")
  #here I also wish to rename the file
else:
  print("Fail")

目标文件名示例：CS123-5_3 (1)

实际输出：

CS123-5_3 (1)
['C']
['S']
['1']
['2']
['3']
['-']
['5']
['_']
['3']
['(']
['1']
[')']
Fail #here fail refers to the if any statement

预期分割结果：['CS123', '-', '5', '_', '3', '(1)']

需要改进代码实现预期分割、匹配判断和文件重命名。

问题分析与改进方案 #

1. 错误的字符遍历：你写的for fname in f_name:会逐个遍历文件名的每个字符，导致每次只对单个字符做分割，自然得到单个字符的列表。直接对整个f_name字符串处理即可，不需要这个循环。
2. 分割方式错误：str.split()不支持正则表达式作为分隔符，也无法保留分隔符。要实现保留分隔符的正则分割，必须用re.split()，并通过捕获组来保留分隔符。
3. 匹配逻辑位置错误：当前的if语句在文件名遍历循环之外，只会处理最后一次的分割结果。应该把匹配逻辑放到遍历循环内部，每个文件单独判断。
4. 文件名处理不严谨：x[:-4]直接截断后缀的写法容易出错（比如后缀不是4个字符的情况），用os.path.splitext()可以正确分离文件名和后缀；另外重命名时必须用完整路径，否则可能找不到文件，要用os.path.join()拼接路径。

改进后的完整代码 #

import pandas as pd
import os
import re

# 读取条件列表，避免用id作为变量名（它是Python内置函数）
df = pd.read_excel(r"C:\Users\Asus\Desktop\py\con.xlsx")
id_list = df.iloc[:, 3].tolist()
file_dir = r"C:\Users\Asus\Desktop\Python\Input"  # 路径加r避免转义问题

# 遍历目录下的所有文件
for filename in os.listdir(file_dir):
    # 正确分离文件名和后缀
    f_name, ext = os.path.splitext(filename)
    print(f"当前处理文件: {f_name}{ext}")
    
    # 正则分割：捕获-或_作为分隔符保留，最多分割5次得到6个元素
    new_fn = re.split(r'(-|_)', f_name, maxsplit=5)
    print(f"分割结果: {new_fn}")
    
    # 检查条件列表中的元素是否存在于分割结果中
    match_found = any(ele in new_fn for ele in id_list)
    if match_found:
        print("匹配成功，执行重命名")
        # 拼接完整路径，避免重命名失败
        old_file_path = os.path.join(file_dir, filename)
        # 自定义重命名规则，示例为【匹配成功_原文件名】，可按需修改
        new_file_name = f"匹配成功_{filename}"
        new_file_path = os.path.join(file_dir, new_file_name)
        os.rename(old_file_path, new_file_path)
        print(f"文件已重命名为: {new_file_name}")
    else:
        print("未匹配到条件")

代码说明 #

• 正则分割逻辑：re.split(r'(-|_)', f_name, maxsplit=5)中，(-|_)是捕获组，会把-或_作为独立元素保留在分割结果里；maxsplit=5表示最多分割5次，最终得到6个元素，完全符合你的预期。
• 路径处理：os.path.splitext()确保正确分离文件名和后缀，os.path.join()拼接完整路径，避免因路径问题导致重命名失败。
• 重命名规则：示例中把匹配成功的文件重命名为匹配成功_原文件名，你可以根据实际需求修改新文件名的格式。

内容的提问来源于stack exchange，提问作者mayliew