python中os.listdir用法
一、 获取指定文件夹下的所有文件
import os
path_name = 'E:\test\' # 输入要获取文件的根目录
for filename in os.lisdir(path_name):
print(filename) # 输出获取的文件名
os.path.isdir()用于判断对象是否为一个目录。
二、 获取指定文件夹下的所有图片
import os
path_name = 'E:\test\' # 输入要获取图片的根目录
for filename in os.lisdir(path_name):
if os.path.splitext(filename)[1] == '.jpg': # 目录下包含的.jpg文件
print(filename) # 输出获取的图片名
os.path.splitext():分离文件名与扩展名
三、 按顺序读取文件夹下的图片
import os
path_name = 'E:\test\' # 输入要获取图片的根目录
path_list = os.listdir(path_name)
path_list.sort(key=lambda x: int(x.split('.')[0])) # 排序
for filename in os.lisdir(path_name):
if os.path.splitext(filename)[1] == '.jpg': # 目录下包含的.jpg文件
print(filename) # 输出获取的图片名
用os.listdir读取到的文件路径是乱序的,调用list.sort()函数可以对文件按顺序排序。