存储数据的方式

常用的存储数据的方式有两种——存储成csv格式文件、存储成Excel文件

什么是CSV？

csv是一种字符串文件的格式，它组织数据的语法就是在字符串之间加分隔符——行与行之间是加换行符，同行字符之间是加逗号分隔。

它可以用任意的文本编辑器打开（如记事本），也可以用Excel打开，还可以通过Excel把文件另存为csv格式（因为Excel支持csv格式文件）。

用csv格式存储数据，读写比较方便，易于实现，文件也会比Excel文件小。但csv文件缺少Excel文件本身的很多功能，比如不能嵌入图像和图表，不能生成公式。

存储数据的基础知识

存储成csv格式文件和存储成Excel文件，这两种不同的存储方式需要引用的模块不同。

6-1.png

csv写入与读取

如何往csv格式文件写入数据？

1、引用csv模块
写法：
import csv#引用csv模块。
csv_file = open('demo.csv','w',newline='',encoding='utf-8')#创建csv文件，我们要先调用open()函数，传入参数：文件名“demo.csv”、写入模式“w”、newline=''、encoding='utf-8'。

说明：加newline=' '参数的原因是，可以避免csv文件出现两倍的行距（就是能避免表格的行与行之间出现空白行）。加encoding='utf-8'，可以避免编码问题导致的报错或乱码。

2、创建完csv文件后，借助csv.writer()函数来建立一个writer对象
写法：
writer = csv.writer(csv_file)# 用csv.writer()函数创建一个writer对象

3、调用writer对象的writerow()方法，往csv文件里写入内容
写法：
writer.writerow(['电影','豆瓣评分'])#借助writerow()函数可以在csv文件里写入一行文字 "电影"和“豆瓣评分”。

特别需要注意：writerow()函数里，需要放入列表参数，所以我们得把要写入的内容写成列表。就像['电影','豆瓣评分']

4、完整的写入CSV代码写法：
import csv#引用csv模块。
csv_file = open('demo.csv','w',newline='',encoding='utf-8')#调用open()函数打开csv文件，传入参数：文件名“demo.csv”、写入模式“w”、newline=''、encoding='utf-8'。
writer = csv.writer(csv_file)# 用csv.writer()函数创建一个writer对象。
writer.writerow(['电影','豆瓣评分'])#调用writer对象的writerow()方法，可以在csv文件里写入一行文字 “电影”和“豆瓣评分”。
writer.writerow(['银河护卫队','8.0'])#在csv文件里写入一行文字 “银河护卫队”和“8.0”。
writer.writerow(['复仇者联盟','8.1'])#在csv文件里写入一行文字 “复仇者联盟”和“8.1”。
csv_file.close()#写入完成后，关闭文件就大功告成啦！

怎么读取csv文件的数据

1、调用和打开CSV模块
import csv
csv_file = open('demo.csv','r',newline='',encoding='utf-8')# 用open()打开“demo.csv”文件，'r'是read读取模式，newline=''是避免出现两倍行距。encoding='utf-8'能避免编码问题导致的报错或乱码。

2、建立一个“朗读者”
reader = csv.reader(csv_file)# 用csv.reader()函数创建一个reader对象

3、遍历读取
for row in reader:
print(row)# 用for循环遍历reader对象的每一行。打印row，就能读取出“demo.csv”文件里的内容

csv模块官方文档链接：
https://yiyibooks.cn/xx/python_352/library/csv.html#module-csv

Excel写入与读取

如何往EXCEL格式文件写入数据

1、 提前安装好openpyxl模块

2、 引用它

3、 通过openpyxl.Workbook()函数创建新的工作薄
写法：
import openpyxl #引用openpyxl
wb = openpyxl.Workbook() #利用openpyxl.Workbook()函数创建新的workbook（工作薄）对象，就是创建新的空的Excel文件

4、 创建完新的工作薄后，获取工作表。
写法：
sheet = wb.active#wb.active就是获取这个工作薄的活动表，通常就是第一个工作表。
sheet.title = 'new title'#可以用.title给工作表重命名。现在第一个工作表的名称就会由原来默认的“sheet1”改为"new title"。

5、 往单元格里写入内容
sheet['A1'] = '漫威宇宙' #把'漫威宇宙'赋值给第一个工作表的A1单元格，就是往A1的单元格中写入了'漫威宇宙'。

6、 如果我们想往工作表里写入一行内容的话，就得用到append函数
row = ['美国队长','钢铁侠','蜘蛛侠']#把我们想写入的一行内容写成列表，赋值给row。
sheet.append(row)#用sheet.append()就能往表格里添加这一行文字。

7、 如果我们想要一次性写入的不止一行，而是多行内容
rows = [['美国队长','钢铁侠','蜘蛛侠'],['是','漫威','宇宙', '经典','人物']]#先把要写入的多行内容写成列表，再放进大列表里，赋值给rows。
for i in rows:
sheet.append(i)#遍历rows，同时把遍历的内容添加到表格里，这样就实现了多行写入。
print(rows)#打印rows

8、保存这个Excel文件
wb.save('Marvel.xlsx')#保存新建的Excel文件，并命名为“Marvel.xlsx”

9、完整的写入EXCEL文件数据的代码：
import openpyxl #调用openpyxl模块
wb=openpyxl.Workbook() #建立工作簿
sheet=wb.active#建立工作表
sheet.title='new title'#给工作表命名
sheet['A1'] = '漫威宇宙'#定义单元格A1的内容
rows= [['美国队长','钢铁侠','蜘蛛侠'],['是','漫威','宇宙', '经典','人物']]#给出其他单元格待填写的内容
for i in rows:
sheet.append(i)#遍历这些内容并逐一写入工作表
print(rows)#打印内容
wb.save('Marvel.xlsx')#保存并关闭内容

怎么读取Excel文件

对应上面写入的EXCEL文件，完整读取它的代码如下：
wb = openpyxl.load_workbook('Marvel.xlsx')# 调用openpyxl.load_workbook()函数，打开“Marvel.xlsx”文件工作簿
sheet = wb['new title']# 获取“Marvel.xlsx”工作薄中名为“new title”的工作表
sheetname = wb.sheetnames
print(sheetname)# sheetnames是用来获取工作薄所有工作表的名字的。A1_cell = sheet['A1']#
A1_value = A1_cell.value
print(A1_value)# 把“new title”工作表中A1单元格赋值给A1_cell，再利用单元格value属性，就能打印出A1单元格的值
openpyxl模块的官方文档：https://openpyxl.readthedocs.io/en/stable/