python操作excel的函数编写
时间: 2023-08-31 16:10:41 浏览: 91
你可以使用Python中的openpyxl库来操作Excel文件。下面是一个示例代码,演示如何使用openpyxl库读取和写入Excel文件。
首先,你需要安装openpyxl库。你可以使用以下命令来安装:
```
pip install openpyxl
```
接下来,你可以使用以下代码来读取Excel文件的内容:
```python
from openpyxl import load_workbook
def read_excel(file_path, sheet_name):
# 加载Excel文件
workbook = load_workbook(file_path)
# 选择工作表
sheet = workbook[sheet_name]
# 读取数据
data = []
for row in sheet.iter_rows(values_only=True):
data.append(row)
# 关闭Excel文件
workbook.close()
return data
```
在上面的代码中,`file_path`是Excel文件的路径,`sheet_name`是要读取的工作表的名称。函数`read_excel`会返回一个二维数组,其中每一行代表Excel文件中的一行数据。
接下来,你可以使用以下代码来写入Excel文件:
```python
from openpyxl import Workbook
def write_excel(file_path, sheet_name, data):
# 创建一个新的Excel工作簿
workbook = Workbook()
# 获取默认的工作表
sheet = workbook.active
# 设置工作表名称
sheet.title = sheet_name
# 写入数据
for row in data:
sheet.append(row)
# 保存Excel文件
workbook.save(file_path)
# 关闭Excel文件
workbook.close()
```
在上面的代码中,`file_path`是要保存的Excel文件的路径,`sheet_name`是要创建的工作表的名称,`data`是要写入的数据,它应该是一个二维数组,其中每一行代表Excel文件中的一行数据。
这只是一个简单的示例,openpyxl库还提供了许多其他功能,例如样式设置、合并单元格等。你可以参考openpyxl的官方文档来了解更多详细信息:https://openpyxl.readthedocs.io/
阅读全文