如何搜索文件有一个已知的文件扩展名，如.py？

Question

如何搜索文件有一个已知的文件扩展名，如.py ??

fext = raw_input("Put file extension to search: ")
dir = raw_input("Dir to search in: ")

##Search for the file and get the right one's

Answer 1

我相信你想做类似的事情：/ /dir/to/search/*.extension /to/ /dir/to/search/*.extension . /dir/to/search/*.extension ？

这叫做glob，以下是如何使用它：

import glob
files = glob.glob('/path/*.extension')

编辑：这里是文档： http ： //docs.python.org/library/glob.html

Answer 2

import os
root="/home"
ext = raw_input("Put file extension to search: ")
path = raw_input("Dir to search in: ")
for r,d,f in os.walk(path):
    for files in f:
         if files.endswith(ext):
              print "found: ",os.path.join(r,files)

Answer 3

非递归：

for x in os.listdir(dir):
    if x.endswith(fext):
        filename = os.path.join(dir, x)
        # do your stuff here

Answer 4

你可以写的很简单：

import os
ext = raw_input("Put file extension to search: ")
path = raw_input("Dir to search in: ")
matching_files = [os.path.join(path, x) for x in os.listdir(path) if x.endswith(ext)]