输入路径,寻找某个格式结尾的文件

1.使用python模块查找(格式自己转换)

'''

os.walk(path)得到一个generator对象

可以使用for遍历

每次遍历会得到一个元组,这个元组有三个值

dirpath: 当前操作的路径

dirnames: 当前路径下的所有文件夹组成的列表

filenames: 当前路径下的所有文件组成的列表

'''

import os

def findFile(path,suffix):

for dirpath,dirnames,filenames in os.walk(path):

for fileName in filenames:

if fileName.endswith(suffix):

print(os.path.join(dirpath,fileName))

def main():

path = input('请输入文件夹路径:')

suffix = input('请输入后缀名(带.):')

findFile(path,suffix)

main()

2.使用遍历方法查找

import os

import os.path #导入所需模块

#os.listdir:返回指定目录下所有文件和目录名

def findFile(path,suffix):

'''

找到path下所有的以suffix结尾的文件,

path:路径

suffix:后缀

'''

ls = os.listdir(path)  #将指定路径下的所有文件和目录名放入一个列表中

for temp in ls:  #设一个文件名在列表中

tempPath = os.path.join(path,temp) #连接路径和文件名

if os.path.isfile(tempPath):  #判断tempPath是否为一个文件,不存在是也返回false

if temp.endswith(suffix):  #

fileLists.append(tempPath)

else:

findFile(tempPath,suffix)

def main():

global fileLists

fileLists = []

path = input ('请输入文件夹路径:')

suffix = input ('请输入后缀名(带.):')

if os.path.isdir(path):

findFile(path,suffix)

print(fileLists)

else:

print('输入文件夹不存在')

main()

你可能感兴趣的:(输入路径,寻找某个格式结尾的文件)