python識別文件格式的方法

小編給大家分享一下python識別文件格式的方法，希望大家閱讀完這篇文章后大所收獲，下面讓我們一起去探討方法吧！

話不多說，直接上代碼：

"""
自動識別 文本編碼格式
"""
import chardet


def detectCode(path):
    with open(path, 'rb') as file:
        data = file.read(20000)
        dicts = chardet.detect(data)
    return dicts["encoding"]


def print_data_1(path):
    """
    這種編碼通過命令行 file -i 文件名獲取編碼格式，
    通過測試，使用file 命令獲取的編碼格式不能獲取正確的編碼數據
    :param path:
    :return:
    """
    with open(path, "r", encoding="iso-8859-1") as f:
        i = 0
        for line in f:
            print(line)
            i += 1
            if i == 5:
                break
    f.close()


def print_data_2(path):
    print("-------------------------------")
    with open(path, "r", encoding="{0}".format(detectCode(path))) as f:
        i = 0
        for line in f:
            b_line = line.encode("utf-8")  # 將文件內容轉化為utf-8格式
            print(chardet.detect(b_line)['encoding'])  # 輸出轉化為內容格式
            i += 1
            if i == 5:
                break
    f.close()


if __name__ == '__main__':
    path = "test.txt"
    print(detectCode(path))
    # print_data_1(path)
    print_data_2(path)

看完了這篇文章，相信你對python識別文件格式的方法有了一定的了解，想了解更多相關知識，歡迎關注億速云行業資訊頻道，感謝各位的閱讀！