高效使用python之csv文件读取和写入

 

CSV后缀的文件是标准文件格式，可以通过文本编辑器或者excel表格打开，

使用非常广泛；使用文本编辑器打开后，每一行都以英文逗号隔开，如下所示：

python安装后默认带了csv库，通过该库我们来介绍下csv文件的读取和写入。csv库的读取和写入函数传递的是文件对象，因此csv库一般要与open函数结合在一起使用。

csv文件读取为列表

打开名称为1.csv的文件，使用csv.reader()函数，读取文件对象，可以对该对象循环打印读取到的内容。

with open('1.csv','r') as f:
	csv_read = csv.reader(f)
	print(csv_read)
	#循环打印
	for i in csv_read:
		print(i)

打印结果：每一行都是一个列表

<_csv.reader object at 0x10d672810>

['买进量', '买进价格']

['100', '98.5']

['200', '99.2']

['100', '97.5']

['300', '98.1']

['200', '99.4']

csv文件读取为字典

打开名称为1.csv的文件，使用csv.DictReader()函数，读取文件对象，可以对该对象循环打印读取到的内容。

with open('1.csv','r') as f:
	csv_read = csv.DictReader(f)
	csv_read.fieldnames = ['a','b']
	print(csv_read)
	
	for i in csv_read:
		print(i)

csv_read.fieldnames = ['a','b'] 表示设置每一行数据对应的字典的key值，如果不设置，会使用csv第一行的内容作为字典的key

打印结果：每一行都是一个字典

{'a': '买进量', 'b': '买进价格'}

{'a': '100', 'b': '98.5'}

{'a': '200', 'b': '99.2'}

{'a': '100', 'b': '97.5'}

{'a': '300', 'b': '98.1'}

{'a': '200', 'b': '99.4'}

因为csv中第一行是标题行，可作为字典的key，此处我们不设置csv_read.fieldnames

with open('1.csv','r') as f:
	csv_read = csv.DictReader(f)
	print(csv_read)
	for i in csv_read:
		print(i)

执行后打印如下：

{'买进量': '100', '买进价格': '98.5'}

{'买进量': '200', '买进价格': '99.2'}

{'买进量': '100', '买进价格': '97.5'}

{'买进量': '300', '买进价格': '98.1'}

{'买进量': '200', '买进价格': '99.4'}

将字典对象写入csv文件

定义一个列表，每个元素为字典。

打开名称为2.csv的文件（本地没有该文件，会打开一个新文件），使用csv.DictWriter()函数，参数fieldnames定义字典的key，通过writeheader函数写入csv文件的标题头，通过writerows函数写入列表中的所有字典对象的value值。

stock_dict= [{'买进量': '100', '买进价格': '98.5'},
             {'买进量': '200', '买进价格': '99.2'},
             {'买进量': '100', '买进价格': '97.5'},
             {'买进量': '300', '买进价格': '98.1'},
             {'买进量': '200', '买进价格': '99.4'}]
with open('2.csv','w') as f:
	csv_write = csv.DictWriter(f,fieldnames=('买进量','买进价格'))
	csv_write.writeheader()
	csv_write.writerows(stock_dict)

执行后，生成2.csv文件，用excel表格打开内容如下：

将列表对象写入csv文件

定义一个列表，每个元素也为为列表。

打开名称为3.csv的文件（本地没有该文件，会打开一个新文件），使用csv.writer()函数，通过writerow函数写入标题头，通过writerows函数写入列表中的所有列表对象。

import csv
stock_list= [[100,98.5],[200,99.2],
             [100,97.5],[300,98.1],
             [200,99.4]]

with open('3.csv','w') as f:
	csv_write = csv.writer(f)
	csv_write.writerow(('买进量','买进价格'))
	csv_write.writerows(stock_list)

执行后，生成3.csv文件，用excel表格打开内容如下：

----感谢读者的阅读和学习，谢谢大家。

共勉： 东汉·班固《汉书·枚乘传》：“泰山之管穿石，单极之绠断干。水非石之钻，索非木之锯，渐靡使之然也。”

-----指水滴不断地滴，可以滴穿石头；

-----比喻坚持不懈，集细微的力量也能成就难能的功劳。