技术 | 如何利用Pycharm读写CSV文件

大家好,我是Lucy@Fintech社区。

在量化交易中,最基础的是什么?当然是数据!这篇文章,我们将会展现如何读取以及将数据写入csv文件

 

导 言

数据有各种各样的存储格式,csv、txt、xml、dat 等等,其中,大家最熟悉的当属excel文件。数据一拿到手,不少人就会熟练地点开Excel,“文件”→“打开”→“确定”,等会儿再用python读取一看,就出现了这样那样的格式问题,比如:日期格式被excel篡改了、中文字符乱码等。

 

因此,比较好的选择是直接把数据存储成csv格式,这样各种文本编辑器都可以直接读取,更重要的是,做量化交易时用得很多的Pycharm编程环境也能很好地读取csv格式文件。下面,我们将展示如何读取及写入csv文件。

 

1. 读取csv文件

前期工作:在定义的py文件里边创建一个excel文件,并另存为csv文件,命名为"test_csv",我们随意放入几行数据,如下:


 


 

先在python环境里导入csv模块:

 

然后定义一个csv文件的变量test_csv,用open把该文件打开,模式采用‘r’(只读)。


 

 

然后进行遍历打印,就可以看到读取的数据了。
 

 

是不是很简单呢?除此之外,我们还可以在Pycharm中安装csv插件,来直接打开csv文件(在Pycharm中搜索“Plugins”,然后搜索关键词“csv”,就可以下载了;安装完成之后记得重启Pycharm让插件生效)。

 

2. 写入csv文件

前期工作:在开始前我们定义三组数据,分别对应Name、Age和Region,例如:

 


 

写入的第一步同样也是打开文件,因为要写入,所以用的模式是  'a'  (只写),"newline="表示csv文件的类型,是一个固定的语句。
 

 

 

下面我们定义一个变量进行写入,将刚才的文件变量传进来,dialect就是定义一下文件的类型,我们定义为excel类型。

 

 

然后进行数据的写入,写入的方法是writerow,通过写入模式对象,调用方法进行写入。

 

 

 最后就用我们最熟悉的一个语句收尾:
 

回头打开csv文件'person_csv2.csv',查看写入结果:
 

 

学会了这些,是不是再也不用担心数据的读写问题啦~