测试人员使用python csv库构造csv测试数据

测试时经常会构造测试数据，分享csv数据构造的方法
首先导入csv库，import csv

• 读取文件
  1、使用csv.reader() 读取 csv 文件内容

import csv  # 模块导入

with open('aa.csv', 'r', newline='', encoding='utf-8') as f:
	# <_csv.reader object at 0x00000000020918D0>
    reader = csv.reader(f)
    for r in reader:
        print(r)

注意遇到编码报错问题解决

执行报错（UnicodeDecodeError: ‘utf-8’ codec can’t decode byte 0xd0 in position 0: invalid continuation byte）
原因：新建的csv文件问题，是不是把xls文件更改的文件名？
解决这个问题1：另存为csv文件
解决问题2：删除文件重新新建，先新建一个记事本，在更改为csv文件即可
解决问题3：以上两种方式还未解决，因为文件内容编码是非法utf-8格式，通过使用适当的错误处理器，可以处理解码过程中出现的错误字节。常用的错误处理器有​​replace​​、​​ignore​​、​​xmlcharrefreplace​​等，这样做会导致解码过程中出现错误的字节被忽略掉

with open(‘aaa.csv’, ‘r’, newline=‘’,encoding=‘utf-8’, errors=‘ignore’) as f:

或者用​​gbk​​编码解码文件：（文件中有汉字，编码可以替换成gbk）
pythonCopy codewith open(‘file.txt’, ‘r’, encoding=‘gbk’) as file:
content = file.read()

2、 使用 DictReader 类读取，返回的结果放到了一个字典的值内，字典的键就是单元格的标题。

import csv  # 模块导入

with open('aa.csv', 'r', newline='', encoding='utf-8') as f:
    reader = csv.DictReader(f)
    for r in reader:
        print(r)

3、通过pandas的read_csv函数读取文件

na_values的用法:将值替换成NaN
比如把null替换成NaN
na_values(nulll)

file_csv = pd.read_csv (file_name, sep=',', encoding='utf-8', engine='python', na_values='NaN', keep_default_na=False)

• 写入文件
  1、创建文件并写入数据，参数w+,会先判断文件是否存在，如果存在写入时会把原有数据替换，如果文件不存在会新建一个新文件并数据写入

with open('aaa.csv',w+',encoding='gbk')as file_obj:
......

2、保留原有数据，追加新数据

with open('aaa.csv','a+',encoding='gbk')as file_obj:
......