python获取文件夹下所有的文件名 os.listdir()

import os

imgs = os.listdir("H:/test1/data1")  #此时是绝对路径，也可以放相对路径

#imgs = os.listdir("./data1")

print(imgs)
['1.png', '2.png', '3.png', '4.png', '5.png', '6.png' ]

listdir（）函数的参数是一个路径名，函数返回值是一个列表，存放该路径下所有的文件名
参考：python中os模块中的listdir（）函数的用法