如何根据字符串内字符对文件名列表排序?解决os.listdir乱序问题
问题描述
现有如下乱序的文件名列表:
['training_tech26.txt', 'training_tech41.txt', 'training_tech68.txt', 'training_tech84.txt', 'training_tech52.txt', 'training_sales17.txt', 'training_sales2.txt', 'training_tech47.txt', 'training_sales23.txt', 'training_sales3.txt', 'training_tech9.txt', 'training_tech12.txt']
需要将其按文件名中的数字进行自然排序(比如实现 training_sales2.txt → training_sales3.txt → training_sales17.txt → training_tech9.txt 这类数字升序的效果)。目前通过 os.listdir() 获取的文件列表是乱序的,文件夹内的文件本身有序,需要对列表做自定义排序。
获取文件列表的代码如下:
tech_dir_path = "/path/to/folder/with/files" res = [] tech_res = os.listdir(tech_dir_path)
解决方案
核心思路是提取文件名中的数字部分,转为整数后作为排序依据,以下是两种常用实现方式:
方法1:按数字全局自然排序
直接提取每个文件名里的数字,以此为key排序:
import os import re def extract_num(filename): # 匹配文件名中的数字序列 num_match = re.search(r'\d+', filename) return int(num_match.group()) if num_match else 0 tech_dir_path = "/path/to/folder/with/files" tech_res = os.listdir(tech_dir_path) # 按数字从小到大排序 sorted_files = sorted(tech_res, key=extract_num) print(sorted_files)
排序结果示例:
['training_sales2.txt', 'training_sales3.txt', 'training_tech9.txt', 'training_tech12.txt', 'training_sales17.txt', 'training_tech26.txt', 'training_sales23.txt', 'training_tech41.txt', 'training_tech47.txt', 'training_tech52.txt', 'training_tech68.txt', 'training_tech84.txt']
方法2:先按类别(sales/tech)再按数字排序
如果需要先区分sales和tech类别,同类别内再按数字升序,可以将排序key设为元组:
import os import re def sort_key(filename): # 提取类别标识 category = 'sales' if 'sales' in filename else 'tech' # 提取数字 num_match = re.search(r'\d+', filename) num = int(num_match.group()) if num_match else 0 return (category, num) tech_dir_path = "/path/to/folder/with/files" tech_res = os.listdir(tech_dir_path) sorted_files = sorted(tech_res, key=sort_key) print(sorted_files)
排序结果示例:
['training_sales2.txt', 'training_sales3.txt', 'training_sales17.txt', 'training_sales23.txt', 'training_tech9.txt', 'training_tech12.txt', 'training_tech26.txt', 'training_tech41.txt', 'training_tech47.txt', 'training_tech52.txt', 'training_tech68.txt', 'training_tech84.txt']
补充说明
os.listdir()返回的顺序由文件系统的存储逻辑决定,并非直观的数字/字母序,因此必须通过自定义排序逻辑来实现需求中的有序排列。
内容的提问来源于stack exchange,提问作者Farkhad
相关产品推荐
相关产品推荐

