仅从python中的文件夹中提取特定文件作为列表

import os

def extract_files(folder_path, file_extension):
    file_list = []
    for file in os.listdir(folder_path):
        if file.endswith(file_extension):
            file_list.append(file)
    return file_list

folder_path = '/path/to/folder'
file_extension = '.txt'
files = extract_files(folder_path, file_extension)
print(files)