Python读取tsv文件数据

最新推荐文章于 2024-11-02 13:42:10 发布

蛙鳜鸡鹳狸猿

最新推荐文章于 2024-11-02 13:42:10 发布

阅读量1w

点赞数 2

分类专栏： # Python 文章标签： Python tsv文件 \t Tab 读取

本文链接：https://blog.youkuaiyun.com/sweeper_freedoman/article/details/88802140

版权

Python 专栏收录该内容

32 篇文章

订阅专栏

本文介绍了一个简单的Python程序，用于处理TSV文件。该程序能够读取TSV文件，并将每行数据转换为字典格式，方便进行后续的数据处理和分析工作。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

#!/usr/bin/env python3
# -*- coding: utf-8 -*-


"""
create_author : 蛙鳜鸡鹳狸猿
create_time   : 2019-03-19
program       : *_* .tsv file handler *_*
"""

import codecs


class TSV(object):
    """
    .tsv file's handler.
    """

    def __init__(self, file):
        """
        TSV init.
        :param file: .tsv file to handle.
        """
        self.file = file

    def __repr__(self):
        return "File {file} under handling......".format(file=self.file)

    def tsv(self):
        """
        .tsv file's column definition and data check.
        :return: List.
            lines data from [file] row by row in dict format.
        """
        with codecs.open(self.file, 'r', "utf-8") as f:
            line = f.readline()
            data = []
            head = []
            while line:
                if line.isspace():
                    line = f.readline()
                    continue
                elif not line.isspace():
                    # to be compatible between OS
                    head = line.rstrip("\r\n").split('\t')
                    line = f.readline()
                    break
            while line:
                if line.isspace():
                    line = f.readline()
                    continue
                elif not line.isspace():
                    body = line.rstrip("\r\n").split('\t')
                    rows = zip(head, body)
                    tsv_dic = {}
                    for (head_sub, body_sub) in list(rows):
                        tsv_dic[head_sub] = body_sub
                    data.append(tsv_dic)
                    line = f.readline()
            return data


if __name__ == "__main__":
    with codecs.open("tsv", 'w', "utf-8") as f:
        rows = """


        Id\tContent
        1\tContent1
        2\tContent2
        3\tContent3
        4\tContent4


        1024\tContent1024

        """
        f.writelines(rows.replace(' ', ''))
    TSV_Tester = TSV(file="tsv")
    print(TSV_Tester.tsv())