Python通过最后一个字母和扩展名选择文件

Question

嗨，我在文件夹中有name_E.mat和name_N.mat文件。如何按最后一个字母和扩展名选择它们？ 以下代码是我尝试的：

filedir = r'C:\Users\320037415\Documents\Depth\Proefpersonen\Sub70'
enterprise = glob.glob(filedir + "/_E.mat")
neolead = glob.glob(filedir + "/_N.mat")
filelist = neolead + enterprise

感谢进阶，我是Python新手：）

Answer 1

您可以将*用作文件名的通配符：

enterprise = glob.glob(filedir + "/*_E.mat")
neolead = glob.glob(filedir + "/*_N.mat")

Answer 2

使用os模块。

演示：

import os

filedir = r'C:\Users\320037415\Documents\Depth\Proefpersonen\Sub70'
filelist = [file for file in os.listdir(filedir) if file.endswith(("_E.mat", "_N.mat")) ]