我想设计一个函数如下:
explore("jpg", "../test_media")
outputsample:
{'../test_media':1,'../test_media/travel_photos':3}
也就是说,函数的输出应该是一个字典,其中每个键是文件夹的地址,每个值是该文件夹中直接具有该扩展名的文件数.如果文件夹中不包含我们想要的文件,则不应在字典中.
我自己写的代码:
import os
def explore(ttype, address):
list_dir = list(os.walk(address))
directoirs = dict()
for x in list_dir:
count = 0
if x[2]:
for y in x[2]:
t = y.split('.')
if t[1].lower() == ttype:
count += 1
if count:
directoirs[str(x[0])] = count
return directoirs
但它不能提供所需的输出