Pyhon读取CSV文件

在执行自动化的测试时，需要准备一些测试数据作为控件的输入内容，我们可以把这些数据存在文件中。比如CSV格式的文件（Comma Separated Values），它是一种以逗号分隔数值的文件类型。在数据库或电子表格中，是最常见的导入导出格式，它以一种简单而明了的方式存储和共享数据，CSV文件通常以纯文本的方式存储数据表。如下表：

 Python通过导入CSV库，可以对CSV文件进行读取操作：

代码示例：

# 导入CSV模块import csv

class FileManager:

    def get_data(self, file_dir):
        """
        以读的方式打开CSV文件，放在mydata列表并返回mydata
        :return:
        """
        with open(file_dir, "r") as file:  # 以读的方式打开文件，文件句柄名命为“file”
            data = csv.reader(file)  
            my_data = []
            for line in data:
                """
                遍历csv每一行数据（每行数据以列表形式存储），追加到my_data列表
                """
                my_data.append(line)
        return my_data


if __name__ == '__main__':
    f = FileManager()
    info = f.get_data("D:\员工信息.csv")
    print(info)

输出结果如下图，我们可以把这个结果放在一个变量里（如通过ddt数据驱动，批量执行测试用例），需要的时候就可以拿到里面的数据。

附：文件处理操作状态

'r'：读
'w'：写
'a'：追加
'r+' == r+w（可读可写，文件若不存在就报错(IOError)）
'w+' == w+r（可读可写，文件若不存在就创建）
'a+' ==a+r（可追加可写，文件若不存在就创建）

如果是二进制文件，就都加一个b：‘rb'　　'wb'　　'ab'　　'rb+'　　'wb+'　　'ab+'