1. 程式人生 > >python遍歷資料夾取到檔案並輸出每種檔案有幾個

python遍歷資料夾取到檔案並輸出每種檔案有幾個

1. 遍歷某個資料夾下的所有檔案,並統計各個檔案型別的個數。
import os#引入os模組

def lister(curr_dir):
stack=[curr_dir]
x = 0#定義mp4檔案
y = 0#定義txt檔案
while stack:
dir=stack.pop()
for path in os.listdir(dir):
file_path=os.path.join(dir,path)
if os.path.isdir(file_path):
stack.append(file_path)
else:
print(path)#輸出每個檔案
a=path.split(".")#取出字尾名
for i in a:#遍歷取到字尾名
if i==‘mp4’:#得到mp4檔案個數
x+=1
elif i==“txt”:#得到txt檔案個數
y+=1
print(‘其中有視訊%s部,小說%s本’ % (x, y))
if name

==‘main’:
lister(‘data’)