python使用pandas操作xlsx

豆芽脚脚2024-05-26 17:08

python操作xlsx有很多种方法，以前使用其他控件操作，使用这个pandas之后发现更好用。场景，我需要读取xlsx模板，然后根据模板去获取数据，根据用户要求导出指定的xlsx文件。

读取文件

cpp 复制代码

data = pd.read_excel('sleepStageAhi.xlsx')
head = data.columns.tolist() # 表格头
cloum =data.values.tolist()[0] #模板字段

写文件

cpp 复制代码

list = []
#连数据库，查询数据
db =MySqLHelper()
sql1 = ""
result1 = db.selectall(sql1)
if len(result1[1]) > 0:
    for item in result1[1]:
        bak = saveMongo(item['zid'],cloum) # 连接mongo获取数据
        if bak != None: # 删除空数据记录
            list.append(bak)
# list转换为DataFrame类型数据            
df = pd.DataFrame(list,columns=head)
#写文件
df.to_excel("export.xlsx",sheet_name="Sheet1",index=True)