溫馨提示×

怎么用python統(tǒng)計(jì)文件類型及數(shù)量

小億
128
2023-10-21 11:00:14
欄目: 編程語言

要統(tǒng)計(jì)文件類型及數(shù)量,你可以使用Python的os模塊來遍歷文件夾中的文件,然后使用文件的后綴來判斷文件類型。以下是一個示例代碼:

import os

def count_file_types(folder_path):
    file_types = {}

    for root, dirs, files in os.walk(folder_path):
        for file in files:
            file_name, file_ext = os.path.splitext(file)
            if file_ext:
                file_type = file_ext[1:].lower()
                if file_type in file_types:
                    file_types[file_type] += 1
                else:
                    file_types[file_type] = 1

    return file_types

folder_path = 'path/to/folder'
file_types = count_file_types(folder_path)

for file_type, count in file_types.items():
    print(f"{file_type}: {count}")

在上述示例代碼中,count_file_types函數(shù)接受一個文件夾路徑作為參數(shù),使用os.walk遍歷該文件夾中的所有文件。對于每個文件,使用os.path.splitext函數(shù)來獲取文件的文件名和后綴。然后,將后綴轉(zhuǎn)換為小寫,并使用字典file_types來記錄每個文件類型的數(shù)量。最后,遍歷file_types字典,輸出每個文件類型及其對應(yīng)的數(shù)量。

請注意,上述代碼只會統(tǒng)計(jì)文件的后綴作為文件類型。如果你需要更精確的文件類型信息,可以使用第三方庫如mimetypes來識別文件的MIME類型。

0