测试时经常会构造测试数据,分享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:
......