Python对文件名进行分类

Python是一种高级编程语言，提供了丰富的库和功能，方便开发人员对文件进行操作和管理。在Python中，我们可以使用不同的方法来对文件名进行分类，以便更好地组织和管理文件。本文将从多个方面对Python对文件名进行分类进行详细的阐述。

一、文件类型分类

1、根据文件扩展名分类

在许多情况下，我们可以通过文件的扩展名来判断文件类型。Python提供了os模块的splitext函数，可以方便地获取文件的扩展名。我们可以使用字典来存储不同类型文件的列表，并将文件根据扩展名进行分类。

import os

def classify_by_extension(file_list):
    file_dict = {}
    for file in file_list:
        filename, extension = os.path.splitext(file)
        if extension in file_dict:
            file_dict[extension].append(file)
        else:
            file_dict[extension] = [file]
    return file_dict

file_list = ["file1.txt", "file2.jpg", "file3.txt", "file4.py", "file5.doc"]
file_dict = classify_by_extension(file_list)
print(file_dict)
# 输出：{'.txt': ['file1.txt', 'file3.txt'], '.jpg': ['file2.jpg'], '.py': ['file4.py'], '.doc': ['file5.doc']}

2、根据文件类型分类

除了根据扩展名分类外，我们还可以根据文件的内容来判断文件类型。通过读取文件的二进制数据，可以根据文件的特征进行分类。例如，我们可以根据文件的魔术数字（Magic Number）来判断文件类型。

import magic

def classify_by_type(file_list):
    file_dict = {}
    for file in file_list:
        file_type = magic.from_file(file, mime=True)
        if file_type in file_dict:
            file_dict[file_type].append(file)
        else:
            file_dict[file_type] = [file]
    return file_dict

file_list = ["file1.txt", "file2.jpg", "file3.txt", "file4.py", "file5.doc"]
file_dict = classify_by_type(file_list)
print(file_dict)
# 输出：{'text/plain': ['file1.txt', 'file3.txt'], 'image/jpeg': ['file2.jpg'], 'text/x-python': ['file4.py'], 'application/msword': ['file5.doc']}

二、文件大小分类

1、根据文件大小范围分类

有时候，我们希望将文件根据大小进行分类，例如将文件分为小于1MB、1MB到10MB和大于10MB三个类别。在Python中，我们可以使用os模块的getsize函数来获取文件的大小，并根据文件大小范围进行分类。

import os

def classify_by_size(file_list):
    file_dict = {
        'Small Files': [],
        'Medium Files': [],
        'Large Files': []
    }
    for file in file_list:
        file_size = os.path.getsize(file)
        if file_size < 1024 * 1024:
            file_dict['Small Files'].append(file)
        elif file_size < 10 * 1024 * 1024:
            file_dict['Medium Files'].append(file)
        else:
            file_dict['Large Files'].append(file)
    return file_dict

file_list = ["file1.txt", "file2.jpg", "file3.txt", "file4.py", "file5.doc"]
file_dict = classify_by_size(file_list)
print(file_dict)
# 输出：{'Small Files': ['file1.txt', 'file3.txt', 'file4.py'], 'Medium Files': ['file2.jpg'], 'Large Files': ['file5.doc']}

2、根据文件大小比例分类

除了简单地根据文件大小范围进行分类外，我们还可以根据文件大小占总文件大小的比例进行分类。例如，我们可以将文件分为小于10%、10%到50%和大于50%三个类别。通过计算文件大小占比，可以更细粒度地对文件进行分类。

import os

def classify_by_ratio(file_list, total_size):
    file_dict = {
        'Less than 10%': [],
        'Between 10% and 50%': [],
        'Greater than 50%': []
    }
    for file in file_list:
        file_size = os.path.getsize(file)
        file_ratio = file_size / total_size
        if file_ratio < 0.1:
            file_dict['Less than 10%'].append(file)
        elif file_ratio < 0.5:
            file_dict['Between 10% and 50%'].append(file)
        else:
            file_dict['Greater than 50%'].append(file)
    return file_dict

file_list = ["file1.txt", "file2.jpg", "file3.txt", "file4.py", "file5.doc"]
# 假设总文件大小为100MB
total_size = 100 * 1024 * 1024
file_dict = classify_by_ratio(file_list, total_size)
print(file_dict)
# 输出：{'Less than 10%': ['file1.txt', 'file4.py'], 'Between 10% and 50%': ['file2.jpg'], 'Greater than 50%': ['file5.doc']}

三、其他分类方式

除了文件类型和文件大小，我们还可以根据其他方式对文件名进行分类。

1、根据文件名中的关键词分类

通过分析文件名中的关键词，我们可以将文件按照不同的主题进行分类。例如，将文件名中包含"python"关键词的文件归类到"Python相关"，将文件名中包含"project"关键词的文件归类到"项目文件"。

def classify_by_keyword(file_list, keyword):
    file_dict = {
        'Related to ' + keyword: []
    }
    for file in file_list:
        if keyword in file:
            file_dict['Related to ' + keyword].append(file)
    return file_dict

file_list = ["file1.txt", "file2.jpg", "file3.txt", "file4.py", "file5.doc"]
keyword = "python"
file_dict = classify_by_keyword(file_list, keyword)
print(file_dict)
# 输出：{'Related to python': ['file4.py']}

2、根据文件名长度分类

有时候，我们希望将文件根据文件名的长度进行分类。例如，将文件名长度小于10个字符的文件归类到"短文件名"，将文件名长度大于20个字符的文件归类到"长文件名"。

def classify_by_length(file_list, length):
    file_dict = {
        'Short File Names': [],
        'Long File Names': []
    }
    for file in file_list:
        if len(file) < length:
            file_dict['Short File Names'].append(file)
        else:
            file_dict['Long File Names'].append(file)
    return file_dict

file_list = ["file1.txt", "file2.jpg", "file3.txt", "file4.py", "file5.doc"]
length = 10
file_dict = classify_by_length(file_list, length)
print(file_dict)
# 输出：{'Short File Names': ['file1.txt', 'file3.txt', 'file4.py'], 'Long File Names': ['file2.jpg', 'file5.doc']}

通过以上的分类方式，我们可以更好地组织和管理文件，提高对文件的查找和处理效率。