python读取某目录下指定后缀的所有文件

import utils
import os
path=YOUR_DIR
path_list=os.listdir(path)
for filename in path_list:
    if os.path.splitext(filename)[1]==".train": 
    	# splitext会把文件名拆成两部分，第一部分是文件名，第二部分是后缀名
    	# 不管有多少个点，都是只把最后一个点以及后面的字符串当作后缀名
    	# 比如文件名为a.b.c，会把"a.b"当作文件名，".c"当作后缀名
        print(filename)

参考文章：
1.https://blog.csdn.net/weixin_36670529/article/details/107208026