【Python】根据文件名中的数字对列表进行排序

我想从一个路径读取数据集:

from os import listdir
import re
dir = listdir('F:/dataset/TNO_test/ir')
print(dir)

结果是:

['1.bmp', '10.bmp', '11.bmp', '12.bmp', '13.bmp', 
可以使用Python内置的sorted()函数进行排序排序时需要自定义排序规则。 假设要按照文件名中的数字进行排序,可以使用正则表达式提取数字作为排序依据。具体实现步骤如下: 1. 使用os模块获取文件列表 ```python import os # 获取指定目录下的所有文件 files = os.listdir('/path/to/dir') ``` 2. 使用正则表达式提取文件名中的数字 ```python import re def extract_number(filename): # 匹配文件名中的数字 match = re.search(r'\d+', filename) if match: return int(match.group()) else: return -1 # 如果文件名中不含数字,则返回-1 # 测试提取函数 print(extract_number('file1.txt')) # 输出1 print(extract_number('file2.txt')) # 输出2 print(extract_number('file.txt')) # 输出-1 ``` 3. 使用sorted()函数进行排序 ```python sorted_files = sorted(files, key=extract_number) ``` 完整代码示例: ```python import os import re def extract_number(filename): # 匹配文件名中的数字 match = re.search(r'\d+', filename) if match: return int(match.group()) else: return -1 # 如果文件名中不含数字,则返回-1 # 获取指定目录下的所有文件 files = os.listdir('/path/to/dir') # 按文件名中的数字排序 sorted_files = sorted(files, key=extract_number) # 输出排序结果 for filename in sorted_files: print(filename) ``` 注意:以上代码只能按照文件名中的数字进行排序,如果要按照其他规则进行排序,需要修改extract_number()函数中的正则表达式。
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

果壳小旋子

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值