在python中列出目录中具有特定扩展名的文件的前几部分

import os

def list_files_with_extension(directory, extension):

    files = os.listdir(directory)

    files_with_extension = [file for file in files if file.endswith(extension)]

    return files_with_extension[:n]

import os

def list_files_with_extension(directory, extension, n):
    files = os.listdir(directory)
    files_with_extension = [file for file in files if file.endswith(extension)]
    return files_with_extension[:n]

directory = '/path/to/directory'
extension = '.jpg'
n = 5

files = list_files_with_extension(directory, extension, n)
for file in files:
    print(file)