Pyhon读取CSV文件

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

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

代码示例：

 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)

View Code

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

附：文件处理操作状态

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

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

Pyhon读取CSV文件

推荐阅读