python创建一个excel,赋值5行5列且带主函数的完整代码
时间: 2023-08-31 13:41:51 浏览: 37
下面是一个完整的Python代码示例,用于创建一个Excel文件并在其中赋值5行5列的数据。代码中包括一个主函数,用于将操作封装。
```python
from openpyxl import Workbook
def create_excel(rows, cols, filename):
# 创建工作簿
workbook = Workbook()
# 获取活动工作表
sheet = workbook.active
# 写入数据
for i in range(1, rows+1):
for j in range(1, cols+1):
cell = sheet.cell(row=i, column=j)
cell.value = i * j
# 保存工作簿
workbook.save(filename)
if __name__ == '__main__':
# 调用主函数
create_excel(5, 5, 'example.xlsx')
```
在这个例子中,我们定义了一个名为`create_excel`的函数,它接受三个参数:行数、列数和文件名。在函数中,我们使用`openpyxl`库来创建一个工作簿,并获取活动工作表。然后,我们使用两个循环来遍历所有的单元格,将相应的乘积写入每个单元格。最后,我们保存工作簿并将其命名为给定的文件名。
在主函数中,我们调用`create_excel`函数,并传入行数、列数和文件名作为参数。这样,当我们运行这个程序时,它将创建一个名为"example.xlsx"的Excel文件,并在其中写入5行5列的数据。
你可以根据需要修改这个例子,比如修改写入的数据或者更改文件名和路径。