在python中使用glob获取文件列表

Question

我在mydir中有一个csv文件列表。我想得到文件名列表。但是，使用glob如下所示将返回一个空列表。

import glob

mydir = "C:\Data"

file_list = glob(mydir + "*.csv")
print('file_list {}'.format(file_list))

Answer 1

尝试^{}：

import os
from fnmatch import fnmatch

mydir = "C:/Data"
file_list = [file for file in os.listdir(mydir) if fnmatch(file, '*.csv')]

print('file_list {}'.format(file_list))

另外，使用RegEx：

import os
import re

mydir = "C:/Data"
file_list = [file for file in os.listdir(mydir) if re.search('.*\.png', file)]  

print('file_list {}'.format(file_list))

---

顺便说一下，^{}是一个模块，您应该像这样使用^{}：

from glob import glob

mydir = "C:/Data"

file_list = glob(mydir + "/*.csv")
print('file_list {}'.format(file_list))