如何在Python中获取目录和子文件夹中的一些(5)文件列表？

import os

def get_file_list(directory, num_files):
    file_list = []
    count = 0

    for root, dirs, files in os.walk(directory):
        for file in files:
            file_path = os.path.join(root, file)
            file_list.append(file_path)
            count += 1

            if count == num_files:
                return file_list

    return file_list

directory = '/path/to/directory'  # 替换为实际的目录路径
num_files = 5

file_list = get_file_list(directory, num_files)
print(file_list)