Python文档数据存储---CSV数据写入与读取

       常用的数据存储介质有文件、关系式数据库和非关系式数据库。文本文档存储适用于具有时效性的数据，入股市行情、商品信息和排行榜信息等，这类数据具有动态变化性质，非特殊要求下，建议存放文件。

1、CSV数据写入

       写入数据分为单行写入和多行写入，对应的函数分别为writerow和writerows，数据写入CSV的代码如下：

import csv
#若存在csv则打开；若不存在则新建
#若不设置newline=''，则每行数据就会隔一行空白行
with open('csv_test.csv','w',newline = '')as csvfile:
    writer = csv.writer(csvfile)
    #单行写入
    writer.writerow(['姓名','年龄','电话'])
    #多行写入
    data = [
        ('小p','18','123456789'),
        ('小y','20','987654321')
    ]
    writer.writerows(data)

---

2、CSV数据读取

       读取CSV文件的函数有reader和DictReader（读取输出全为string类型），reader函数以列表形式返回，DictReader函数以字典的形式返回，字典的键是单元格的标题（列头），字典的值是对应单元格的值。代码如下：

import csv
#获取全部数据
with open('csv_test.csv','r')as csvfile:
    #以列表形式输出
    reader = csv.reader(csvfile)
    #以字典形式输出，第一行作为字典的键
    reader = csv.DictReader(csvfile)
    #以列表生成器的形式表达，当数据过大时，利用此方法可以解决占用太大内存空间的问题
    rows = [row for row in reader]
    print(rows)

      如果需要获取某行数据，就可以循环全部数据，在对每行数据做一个判断，判断是否符合筛选条件，代码如下：

import csv
#获取单行数据
with open('csv_test.csv','r')as csvfile:
    #一列表的形式输出
    reader = csv.reader(csvfile)
    for row in reader:
        if '小p' in row:
            print(row)
    #以字典的形式输出
    reader = csv.DictReader(csvfile)
    for row in reader:
        if row['姓名'] == '小p':
            print(row)