在数据分析中经常需要从csv格式的文件中存取数据以及将数据写书到csv文件中。将csv文件中的数据直接读取为 dict 类型和 DataFrame 是非常方便也很省事的一种做法,以下代码以鸢尾花数据为例。
csv文件读取为dict
代码
-- coding: utf-8 --
import csv
with open(‘E:/iris.csv’) as csvfile:
reader = csv.DictReader(csvfile, fieldnames=None) # fieldnames默认为None,如果所读csv文件没有表头,则需要指定
list_1 = [e for e in reader] # 每行数据作为一个dict存入链表中
csvfile.close()
print list_1[0]
输出
{‘Petal.Length’: ‘1.4’, ‘Sepal.Length’: ‘5.1’, ‘Petal.Width’: ‘0.2’, ‘Sepal.Width’: ‘3.5’, ‘Species’: ‘setosa’}
如果读入的每条数据需要单独处理且数据量较大,推荐逐条处理然后再放入。
多条类型为dict的数据写入csv文件
代码
数据
data = [
{‘Petal.Length’: ‘1.4’, ‘Sepal.Length’: ‘5.1’, ‘Petal.Width’: ‘0.2’, ‘Sepal.Width’: ‘3.5’, ‘Species’: ‘setosa’},
{‘Petal.Length’: ‘1.4’, ‘Sepal.Length’: ‘4.9’, ‘Petal.Width’: ‘0.2’, ‘Sepal.Width’: ‘3’, ‘Species’: ‘setosa’},
{‘Petal.Length