from pathlib import Path

pths = [pth for pth in Path.cwd().iterdir()]

如果是os.listdir()这会返回一个list，文件非常多的时候可能比较耗内存