获取文件夹中指定后缀名的文件名称

代码如下

# -*- coding:utf-8 -*-
'''=================================================
@文件 :   main.py
@说明 :   
@时间 :   2023/7/8 21:47
@作者 :   ***
@版本 :   1.0
===================================================='''

import os


def read_files(folder_path):
    # 获取文件夹中的所有文件名
    file_names = os.listdir(folder_path)

    # 筛选出后缀为.lib的文件名
    lib_files = [file for file in file_names if file.endswith('.lib')]

    return lib_files


def write_to_txt(file_names, output_file):
    with open(output_file, 'w') as file:
        for file_name in file_names:
            file.write(file_name + '\n')


# 指定待读取的文件夹路径
folder_path = 'D:\\VTK-8.2.0-Prefix-Release\\lib'

# 指定输出txt文件的路径
output_file = 'output.txt'

# 读取文件名
lib_files = read_files(folder_path)

# 输出到txt文件
write_to_txt(lib_files, output_file)