计算一个文件夹中 指定类型 文件的数量，python

import os


def count_file_number(filepath, filetype):
    count = 0
    for root, dirname, filenames in os.walk(filepath):
        for filename in filenames:
            if os.path.splitext(filename)[1] == filetype:
                count += 1
    return count

patientPath = '12345'
dicomSlices = count_file_number(patientPath, '.dcm')