Pyhon读取CSV文件
在执行自动化的测试时,需要准备一些测试数据作为控件的输入内容,我们可以把这些数据存在文件中。比如CSV格式的文件(Comma Separated Values),它是一种以逗号分隔数值的文件类型。在数据库或电子表格中,是最常见的导入导出格式,它以一种简单而明了的方式存储和共享数据,CSV文件通常以纯文本的方式存储数据表。如下表:
Python通过导入CSV库,可以对CSV文件进行读取操作:
代码示例:
![](https://images.cnblogs.com/OutliningIndicators/ContractedBlock.gif)
1 # 导入CSV模块import csv 2 3 class FileManager: 4 5 def get_data(self, file_dir): 6 """ 7 以读的方式打开CSV文件,放在mydata列表并返回mydata 8 :return: 9 """ 10 with open(file_dir, "r") as file: # 以读的方式打开文件,文件句柄名命为“file” 11 data = csv.reader(file) 12 my_data = [] 13 for line in data: 14 """ 15 遍历csv每一行数据(每行数据以列表形式存储),追加到my_data列表 16 """ 17 my_data.append(line) 18 return my_data 19 20 21 if __name__ == '__main__': 22 f = FileManager() 23 info = f.get_data("D:\员工信息.csv") 24 print(info)
输出结果如下图,我们可以把这个结果放在一个变量里(如通过ddt数据驱动,批量执行测试用例),需要的时候就可以拿到里面的数据。
附:文件处理操作状态
'r':读
'w':写
'a':追加
'r+' == r+w(可读可写,文件若不存在就报错(IOError))
'w+' == w+r(可读可写,文件若不存在就创建)
'a+' ==a+r(可追加可写,文件若不存在就创建)
如果是二进制文件,就都加一个b:‘rb' 'wb' 'ab' 'rb+' 'wb+' 'ab+'