背景:工作中需要对数据做处理,生成表格,这里小生封装了一个类,可以实现功能哦,希望可以帮助到大家,话不多说,直接上代码了。
# -*- coding: utf-8 -*-
import pandas as pd
class ExportExcel:
def __init__(self, columns, list_data, excel_name):
# 表格列名
self.columns = columns
# 数据列表 多维数据 eg: [[1,2,3],[4,5,6]]
self.list_data = list_data
# 表名
self.excel_name = excel_name
def port_excell(self):
'''
接收多维列表
'''
# 将列表转换为DataFrame
df = pd.DataFrame(self.list_data)
# 设置列名
df.columns = self.columns
# 设置表路径和表名
file_name = 'data/' + self.excel_name + '.xlsx'
# 将DataFrame写入Excel文件
df.to_excel(file_name, index=False)
return True
使用方法:传入三个数据,columns:表头名称,
list_data:表格中的数据,接收的是多维列表,代码中有说明;
excel_name:表名称
使用的时候直接调用类方法即可,便捷好用。