做培训的网站,网站营销公司哪家好,手机网站设计教育类模板,o2o平台运营是什么意思一 生成器的实际使用#xff08;大量数据的导出#xff09;
#分批导出数据然后分批写入excel
import pandas as pd
import openpyxl
from openpyxl.utils.dataframe import dataframe_to_rowsdef execute_query(query):# 假设这是执行 SQL 查询的函数# 返回查询结果passdef …一 生成器的实际使用大量数据的导出
#分批导出数据然后分批写入excel
import pandas as pd
import openpyxl
from openpyxl.utils.dataframe import dataframe_to_rowsdef execute_query(query):# 假设这是执行 SQL 查询的函数# 返回查询结果passdef fetch_products_in_batches(batch_size1000):offset 0while True:query fSELECT * FROM products LIMIT {batch_size} OFFSET {offset}results execute_query(query)if not results:breakyield resultsoffset batch_sizedef export_to_excel(generator, filename):# 创建一个新的 Excel 工作簿wb openpyxl.Workbook()ws wb.active# 写入表头first_batch next(generator)df pd.DataFrame(first_batch)for r in dataframe_to_rows(df, indexFalse, headerTrue):ws.append(r)# 写入后续批次的数据for batch in generator:df pd.DataFrame(batch)for r in dataframe_to_rows(df, indexFalse, headerFalse):ws.append(r)# 保存工作簿wb.save(filename)# 主逻辑
generator fetch_products_in_batches()
export_to_excel(generator, products.xlsx)