Python pandas在原有Excel中追加或重写sheet工作表-Toy模板网

这篇具有很好参考价值的文章主要介绍了Python pandas在原有Excel中追加或重写sheet工作表。希望对大家有所帮助。如果存在错误或未考虑完全的地方，请大家不吝赐教，您也可以点击"举报违法"按钮提交疑问。

情景描述

想在excel中添加几个工作表来保存python得到的同一类型的多个阶段数据变量，非一次性保存，且是多次打开那种方式。
使用xlwt会覆盖掉以前的所有数据（且是通过遍历保存），看了好多帖子发现pandas可以实现我想要的功能，但是没有很直接的方法可以直接使用，且依然会覆盖以及报错等问题，最后发现pd.ExcelWriter()函数有个重要的参数可以直接实现追加工作表的功能。代码和解释如下：

自定义追加变量到excel工作表的函数

// 导包
import os
import numpy as np
import pandas as pd  # 导入模块
from openpyxl import load_workbook  #之前尝试用这个包读取现有的表格，然后再追加,费了时间但是没搞通
def save_variable(filename,sheetname,variable):  #filename后面记得加上文件的后缀 .xls或者.xlsx
    # 重要的是写入函数对象的建立，即if else里的两种情况
    # 如果文件对象不存在，那么就使用默认的模式，即mode=‘w’，会创建一个新的文件
    if not os.path.exists(filename):
        writer = pd.ExcelWriter(filename, engine='openpyxl')   
    else: # 如果文件对象存在，那么就使用追加模式，设置if_sheet_exists='overlay'，此时需要设置mode=‘a’ 
        writer = pd.ExcelWriter(filename, engine='openpyxl', mode='a', if_sheet_exists='overlay')
	
	# 后面是保存数据相关的代码了，
    pd_data = pd.DataFrame(variable)   #将变量转换成df格式
    pd_data.to_excel(writer, sheet_name=sheetname, header=None, index=None)  #不要索引、不要表头
    writer.save()
    writer.close()
    
if __name__ == '__main__':
    #测试同一个工作表进行重写
    a1 = np.array([1,1,1])
    a2 = np.array([2,2,2])
    filename = './test1.xlsx'    #当前文件夹下建立文件  后缀xlsx或者xls均可，具体区别：不知道。
    save_variable(filename,'1',a1)
    save_variable(filename,'1',a2)    #对于只有一行的数组，写进去的是一列，有空再摸索....

    #测试添加新表到现有的excel文件之中
    b = np.array([[1, 2, 1], [1, 3, 6]])
    save_variable(filename,'2',b)