openpyxl 结合 pandas
时间: 2023-11-12 18:58:01 浏览: 121
Python pandas openpyxl笔记汇总.pdf
openpyxl 是一个用于读写 Excel 文件的 Python 库,而 pandas 是一个用于数据处理和分析的 Python 库。这两个库可以结合使用,实现对 Excel 文件的读取、处理和写入。
具体来说,可以使用 openpyxl 中的 load_workbook 方法读取 Excel 文件,然后将其转换为 pandas 中的 DataFrame 对象进行处理。处理完成后,再将 DataFrame 对象转换为 openpyxl 中的 worksheet 对象,最后使用 save 方法将结果写入 Excel 文件。
以下是一个简单的示例代码:
```python
import openpyxl
import pandas as pd
# 读取 Excel 文件
wb = openpyxl.load_workbook('example.xlsx')
ws = wb.active
# 将 worksheet 转换为 DataFrame
data = ws.values
cols = next(data)
df = pd.DataFrame(data, columns=cols)
# 处理 DataFrame
df['new_col'] = df['old_col'] * 2
# 将 DataFrame 转换为 worksheet
ws = openpyxl.utils.dataframe.dataframe_to_rows(df, index=False, header=True)
for r_idx, row in enumerate(ws, 1):
for c_idx, value in enumerate(row, 1):
ws.cell(row=r_idx, column=c_idx, value=value)
# 写入 Excel 文件
wb.save('example.xlsx')
```
阅读全文