python - 需要查找除.dat文件外的文件夹和子文件夹中的文件数

Question

我需要找到文件夹和子文件夹中的文件数，除了 .dat 文件，这是我的代码：

import os

directoryPath = r'C:\Users\\vishns\Documents\Python_Code\Test_Folder'

number_of_files = sum(
    len(files) for r, d, files in os.walk(directoryPath)
)

print(number_of_files)

标签： python

Answer 1

这是一个解决方案：

import os

nfiles = 0
ndat = 0
directory = "C://Users//vishns//Documents//Python_Code//Test_Folder"

for root, dirs, files in os.walk(directory):
    for file in files:
        if file.endswith(".dat"):
            ndat += 1
        else:
            nfiles += 1
        
print('Number of dat files:{}'.format(ndat))
print('Number of relevant files:{}'.format(nfiles))
print('Number of total files:{}'.format(ndat+nfiles))