如何根据文件扩展名过滤文件名？

import os

def filter_files_by_extension(file_list, extension):
    filtered_files = []
    for file_name in file_list:
        if file_name.endswith(extension):
            filtered_files.append(file_name)
    return filtered_files

file_list = os.listdir()
filtered_files = filter_files_by_extension(file_list, '.txt')
print(filtered_files)

const fs = require('fs');

function filterFilesByExtension(fileList, extension) {
    const filteredFiles = [];
    for (const file of fileList) {
        if (file.endsWith(extension)) {
            filteredFiles.push(file);
        }
    }
    return filteredFiles;
}

const fileList = fs.readdirSync('.');
const filteredFiles = filterFilesByExtension(fileList, '.txt');
console.log(filteredFiles);

import java.io.File;
import java.io.IOException;
import java.nio.file.Files;
import java.nio.file.Path;
import java.nio.file.Paths;
import java.util.ArrayList;
import java.util.List;
import java.util.stream.Collectors;

public class FileFilter {
    public static void main(String[] args) throws IOException {
        List<String> fileList = Files.list(Paths.get("."))
                .map(Path::getFileName)
                .map(Path::toString)
                .collect(Collectors.toList());

        List<String> filteredFiles = filterFilesByExtension(fileList, ".txt");
        System.out.println(filteredFiles);
    }

    public static List<String> filterFilesByExtension(List<String> fileList, String extension) {
        List<String> filteredFiles = new ArrayList<>();
        for (String file : fileList) {
            if (file.endsWith(extension)) {
                filteredFiles.add(file);
            }
        }
        return filteredFiles;
    }
}