python 读取当前文件夹下所有后缀为.lib文件

# -*- coding: utf-8 -*-

import os



'''
def file_name(file_dir):
    for root, dirs, files in os.walk(file_dir):
        #print(root)  # 当前目录路径
        #print(dirs)  # 当前路径下所有子目录
        print(files)  # 当前路径下所有非目录子文件

file_name(file_dir)
'''


#输出当前文件夹下所有后缀为.lib的文件
def traverse(f):
    fs = os.listdir(f)
    print(fs)
    for f1 in fs:
        try:
            if f1.split('.')[1] == 'lib' or f1.split('.')[2] == 'lib':
                print(f1)
        except:
            print()

    return

file_dir = 'E:\\ITK\\ITK_install\\lib'
traverse(file_dir)