Python实现一个函数,函数的功能是创建一个Excel表格,构建sheet列表并命名,按行插入数据;传入参数为Excel文件名,sheet表命,插入数据列表。
时间: 2024-09-13 15:14:41 浏览: 42
Python实现将数据库一键导出为Excel表格的实例
5星 · 资源好评率100%
在Python中,你可以使用`openpyxl`库来创建和操作Excel文件。以下是一个简单的函数实现,该函数可以创建一个Excel文件,根据给定的sheet名称创建sheet,并按行插入数据。
首先,确保你已经安装了`openpyxl`库。如果没有安装,可以使用pip进行安装:
```bash
pip install openpyxl
```
接下来是函数的实现:
```python
from openpyxl import Workbook
def create_excel(file_name, sheet_names, data_list):
# 创建一个Excel工作簿
wb = Workbook()
# 创建一个字典来存储sheet名和对应的行数据列表
sheets_data = {sheet_name: [] for sheet_name in sheet_names}
# 将传入的数据列表按照sheet数量分割
for index, data_row in enumerate(data_list):
# 确定数据行应该被插入的sheet
sheet_name = sheet_names[index % len(sheet_names)]
sheets_data[sheet_name].append(data_row)
# 遍历字典,为每个sheet名创建一个sheet,并插入数据
for sheet_name, rows in sheets_data.items():
ws = wb.create_sheet(title=sheet_name)
for row in rows:
ws.append(row) # 将数据行插入到当前sheet
# 保存工作簿到指定的文件名
wb.save(filename=file_name)
print(f"Excel文件 '{file_name}' 已创建并填充数据。")
# 使用示例
create_excel(
file_name='example.xlsx',
sheet_names=['Sheet1', 'Sheet2', 'Sheet3'],
data_list=[
['ID', 'Name', 'Age'],
[1, 'Alice', 25],
[2, 'Bob', 30],
[3, 'Charlie', 35]
]
)
```
这个函数首先创建了一个Excel工作簿,并初始化了一个字典来存储每个sheet名和对应的行数据列表。然后,它将传入的数据列表分割成多个部分,每个部分对应一个sheet。接着,函数遍历字典,为每个sheet名创建一个sheet,并将数据行插入到对应的sheet中。最后,函数将工作簿保存到指定的文件名。
阅读全文