获取文件类型的函数python

可以使用Python的os模块的splitext函数获取文件的扩展名，然后根据扩展名判断文件类型。

示例代码：

import os

def get_file_type(file_path):
    ext = os.path.splitext(file_path)[-1].lower()
    if ext == '.txt':
        return 'text'
    elif ext == '.png' or ext == '.jpg':
        return 'image'
    elif ext == '.mp3' or ext == '.wav':
        return 'audio'
    elif ext == '.mp4' or ext == '.avi':
        return 'video'
    else:
        return 'unknown'

file_path = 'example.txt'
file_type = get_file_type(file_path)
print(file_type)   # 输出：text

在示例代码中，get_file_type函数接受一个文件路径参数，使用os.path.splitext函数获取文件的扩展名，并转换为小写字母。然后根据扩展名判断文件类型，返回对应的类型名称。如果无法识别扩展名，则返回未知类型。

相关问题

Python 获取文件类型

在Python中，你可以使用mimetypes模块来获取文件类型。该模块根据文件扩展名或文件名猜测文件的MIME类型。以下是一个例子：

import mimetypes

filename = 'example.pdf'
mime_type, encoding = mimetypes.guess_type(filename)
print('MIME类型:', mime_type)

输出结果应该类似于：`MIME类型: application/pdf`

如果该函数不能猜测出MIME类型，则返回None。你可以通过查看mimetypes模块的文档来了解更多关于这个模块的用法和限制。

获取文件名的函数python

可以使用os.path库中的basename函数来获取文件名。示例代码如下：

import os

filepath = '/path/to/file.txt'
filename = os.path.basename(filepath)
print(filename)  # 输出：file.txt

如果要去掉文件扩展名，可以使用os.path.splitext函数：

import os

filepath = '/path/to/file.txt'
filename, ext = os.path.splitext(os.path.basename(filepath))
print(filename)  # 输出：file
print(ext)  # 输出：.txt