Python 批量处理特定格式文件

#批量对文件夹下的'.mat'进行处理

def file_name(file_dir,suff):
  L=[]
  for root, dirs, files in os.walk(file_dir):
    for file in files:
      if os.path.splitext(file)[1] == suff:
        L.append(os.path.join(root, file))
  return L

#调用
file_dir='D:\sa'
suff='.mat'
L=[]
L=file_name(file_dir,suff)