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()函数的用法