操作很简单,我都是官方文档找的,写到优快云上算是笔记, 顺便说一句csdn的markdown比印象笔记友好多了,写着也舒服一些,以后笔记就写在优快云吧。 也给大家一个参考,具体的操作还是移步官方文档,文档地址自己找?
Excel文件操作 openpyxl库的使用
稍微提一下,操作excel的库很多,有说xlwt的, 有说pandas的, 我找的openpyxl,操作简单方便些
import os
import openpyxl # 需要安装 pip install openpyxl
def write_to_xlsx(file_name, sheet_name, write_data):
"""
追加保存文件到 excel文件
:param file_name: 文件名 str(不用加后缀)
:param sheet_name: 操作页名称
:param write_data: 保存数据 list类型 追加保存
:return:
"""
if not os.path.exists("./{}.xlsx".format(file_name)):
# 判断是否由此文件 没有就建立Workbook对象
wb = openpyxl.Workbook()
else:
# 有此文件 加载源文件
wb = openpyxl.load_workbook("./{}.xlsx".format(file_name))
# 获取所有的sheet名称
sheet_names = wb.sheetnames
if sheet_name in sheet_names:
# 判断传入 sheet_name名称是否已存在,存在就切换
ws = wb[sheet_name]
else:
# 不存在就新建
ws = wb.create_sheet(sheet_name)
# 添加数据 list类型
ws.append(write_data)
# 保存数据
wb.save("./{}.xlsx".format(file_name))
if __name__ == '__main__':
# 测试
write_to_xlsx("abc", "bbc", ["123", "aaaa", "zzzz"])
write_to_xlsx("abc", "bbc", ["222", "ccca", "zzzz"])
write_to_xlsx("abc", "bbc", ["333", "zzza", "zzzz"])
CSV文件操作
csv模块是python内置的,但是数据以逗号形式存储, 据我所查不支持sheet分页,但是我存数文件一般都用csv文件,方便,简单
import csv
def save_file(file_name, save_data):
"""
保存文件
:param save_data: 保存数据 list类型
:return:
"""
with open("./{}.csv".format(file_name), "a+", newline="", encoding="utf-8") as data_csv:
csv_writer = csv.writer(data_csv, dialect=("excel",))
csv_writer.writerow(save_data)