python-3.x 如何解决函数中的os.walk错误?

kqhtkvqz  于 2023-06-07  发布在  Python
关注(0)|答案(1)|浏览(140)
def getFiles(_root):
    return next(os.walk(_root))[2]

root = "../../../Image/new_m/"
fileList = getFiles(root)  # return file list
fileList = sorted(fileList)

我得到这个错误,每次我试图运行我的函数,这是应该从我的文件夹中获取文件。每次我尝试运行它时,错误都显示StopIteration

5sxhfpxr

5sxhfpxr1#

import os
def getFiles(_root):
    try:
        return next(os.walk(_root))[2]
    except StopIteration:
        return []  # Return an empty list if the iterator is empty

root = "../../../Image/new_m/"
fileList = getFiles(root)
fileList = sorted(fileList)

你可以试试上面的代码。它会解决你的问题。

相关问题