Python 基于 xlsxwriter 实现百万数据导出 excel
追加导出 + 自动切换 sheet
⚠️ excel 中的每个 sheet 最多只能保存 1048576 行数据
# 获取项目的根路径 rootPath
curPath = os.path.abspath(os.path.dirname(__file__))
rootPath = curPath[:curPath.find(你的项目名称 + "/") + len(
你的项目名称 + "/")]
# 临时文件
local_file_path = os.path.join(rootPath, "temp.xlsx")
# 检查并删除现有的临时文件
if os.path.exists(local_file_path):
os.remove(local_file_path)
sheet_number = 1
sheet_name_format = "Sheet_{}"
# 数据量大,导出的数据又包含url的话,会疯狂报警告,大家用不到可以删掉
workbook = xlsxwriter.Workbook(local_file_path, options={'strings_to_urls': False})
table = workbook.add_worksheet(sheet_name_format.format(sheet_number))
# sheet 数据总条数
page_total = 0
# 要写的行
row_number = 1
# 分批导出,每次 100000 条数据
default_limit = 100000
# 分批导出,第 1 页开始
page_number = 1
while True:
# 分批获取数据
data_list = get_data_list(page_number, default_limit) # 你的数据
if len(data_list) == 0:
break
# sheet总条数,0代表第一次写入数据
if page_total == 0:
# 标题
header = [你的标题]
table.write_row(0, 0, header)
# todo 因为我把每个 sheet 控制在了 100万条,就切换下一个 sheet 了。
# todo 如果各位要是玩极限别忘了这里 page_total + 1
for item in data_list:
table.write_row(row_number, 0, list(item.values()))
row_number = row_number + 1
page_total = page_total + len(data_list)
# 自动切换sheet
if page_total >= 1000000:
# 换下一个sheet
sheet_number = sheet_number + 1
table = workbook.add_worksheet(sheet_name_format.format(sheet_number))
# 初始化
page_total = 0
row_number = 1
page_number = page_number + 1
# 关闭
workbook.close()