📜  检查文件是否为txt python(1)

📅  最后修改于: 2023-12-03 15:10:54.194000             🧑  作者: Mango

检查文件是否为txt文件

在Python中,我们可以使用以下方法检查文件是否为txt文件:

import os

def is_txt_file(file_path):
    """
    判断文件是否为txt文件

    Args:
        file_path (str): 文件路径

    Returns:
        bool: 文件是否为txt文件
    """
    return os.path.splitext(file_path)[1].lower() == '.txt'

上述代码中,我们使用了os.path.splitext方法来获取文件名和扩展名,然后判断扩展名是否为.txt,返回结果为True则表示文件是txt文件,否则为其他类型文件。

这个方法的好处是不会读取文件内容,只会检查文件扩展名,因此对于大文件来说比较节省内存和时间。

我们可以通过以下代码测试该方法:

print(is_txt_file('test.txt'))
print(is_txt_file('test.pdf'))

输出结果为:

True
False

以上就是Python中检查文件是否为txt文件的方法。