存储数据的方式
常用的存储数据的方式有两种——存储成csv格式文件、存储成Excel文件
什么是CSV?
csv是一种字符串文件的格式,它组织数据的语法就是在字符串之间加分隔符——行与行之间是加换行符,同行字符之间是加逗号分隔。
它可以用任意的文本编辑器打开(如记事本),也可以用Excel打开,还可以通过Excel把文件另存为csv格式(因为Excel支持csv格式文件)。
用csv格式存储数据,读写比较方便,易于实现,文件也会比Excel文件小。但csv文件缺少Excel文件本身的很多功能,比如不能嵌入图像和图表,不能生成公式。
存储数据的基础知识
存储成csv格式文件和存储成Excel文件,这两种不同的存储方式需要引用的模块不同。
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/