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()函数可以对文件按顺序排序。