python - Python 中的 os.walk 返回一个空列表

Question

我正在尝试使用 os.walk从目录中返回 .xlsx 文件列表，但它返回一个空列表。

import os
import pathlib

working_directory = 'N:/files path'

def find_filenames(path, suffix):
# First save all filenames in a list: file_list:
    file_list = []
    # Gets a list of all validation xlsx files in the folder:
    for subdir, dirs, files in os.walk(path):
        for file in files:
            if file.endswith(suffix):
                # save filesnames in file_list: 
                file_list.append(file)
            return file_list
        
print(find_filenames(pathlib.Path(working_directory), '.xlsx')

它正在打印这个:

[]

当它看起来像这样:

我错过了什么或做错了什么？谢谢!

Answer 1

应该是:

import os
import pathlib

working_directory = 'N:/files path'

def find_filenames(path, suffix):
# First save all filenames in a list: file_list:
    file_list = []
    # Gets a list of all validation xlsx files in the folder:
    for subdir, dirs, files in os.walk(path):
        for file in files:
            if file.endswith(suffix):
                # save filesnames in file_list: 
                file_list.append(file)
    return file_list

问题是你把 return在错误的地方。如果您在问题中输入类似代码，并且第一个文件不以 .xlsx 结尾，它会直接返回 [] !