pandas的读取csv/excel
2021-09-27 本文已影响0人
Rainbow想喝奶茶
Pandas作为一个数据分析工具,功能很强大,在做自动化测试时候当然也可以应用它来读取csv/excel文件,不过我今天的应用比较简单,还需要进一步的学习,只是实现了简单的读取。
安装
首先,自然是需要下载安装pandas,不过它在读取excel类型的文件时也依赖于xlrd,所以最好两个都下载哦。
pip3 install pandas
pip3 install xlrd
应用-读取
首先第一步应用,是要引入,一般都会写为:
import pandas as pd
在读取时,如果是excel文件,应用的是read_excel()方法,如果是csv文件,应用的是read_csv()方法。
data_csv=pd.read_csv('case1.csv')
data_excel=pd.read_csv('case2.csv')
在这种情况下,直接打印data,可以看到得到的是pandas表格型的数据结构DataFrame。
![](https://img.haomeiwen.com/i26677475/24cb1a1a8c9f66c2.png)
但在测试环节,通常更想要的是字典格式的文件,且有时候只想要指定的列,可以进行如下转换。
test_data=[]
for i in data.index.values:
row_data=data.loc[i,['path','method']].to_dict()
test_data.append(row_data)
输出的test_data如下图所示
![](https://img.haomeiwen.com/i26677475/e42c6354bbe72416.png)
今天时间有限,明天细化一下~~
自我记录,有错误欢迎指正~~