【PART】csv 常用代码

最新推荐文章于 2025-03-05 12:00:00 发布

万载志

最新推荐文章于 2025-03-05 12:00:00 发布

阅读量220

点赞数

CC 4.0 BY-SA版权

文章标签： python 开发语言

本文链接：https://blog.youkuaiyun.com/weixin_43411463/article/details/131058839

该文介绍了三个Python类：CSVReader、CSVDictReader和LoadCSV，它们用于读取CSV文件。CSVReader和CSVDictReader使用内置的csv模块，提供迭代器接口在文件结束时能重置文件指针。LoadCSV类则读取数据并将其存储为字典列表。

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

从 unitest-1.4.9 中拿出来的 csv_reader.py 文件

import csv


class CSVReader:
    "Read test data from csv file using an iterator"

    def __init__(self, file):
        try:
            file = open(file)
        except TypeError:
            pass  # "file" was already a pre-opened file-like object
        self.file = file
        self.reader = csv.reader(file)

    def __next__(self):
        try:
            return next(self.reader)
        except StopIteration:
            # reuse file on EOF
            self.file.seek(0, 0)
            return next(self.reader)


class CSVDictReader:
    "Read test data from csv file using an iterator"

    def __init__(self, file):
        try:
            file = open(file)
        except TypeError:
            pass  # "file" was already a pre-opened file-like object
        self.file = file
        self.reader = csv.DictReader(file)

    def __next__(self):
        try:
            return next(self.reader)
        except StopIteration:
            # reuse file on EOF
            self.file.seek(0, 0)
            next(self.reader)  # skip header line
            return next(self.reader)


class LoadCSV:

    def __init__(self, file):
        try:
            file = open(file)
        except TypeError:
            pass  # "file" was already a pre-opened file-like object
        self.file = file
        self.reader = csv.DictReader(file)

    def read_data(self):
        data = []
        for element in self.reader:
            data.append(element)

        return data