python excel 填充颜色_pandas to_excel 添加颜色操作
时间: 2023-10-08 18:05:26 浏览: 563
pandas to_excel 添加颜色操作
在使用 pandas 的 to_excel 方法将 DataFrame 保存为 Excel 文件时,可以通过设置 pandas.ExcelWriter 对象的 engine 参数为 "openpyxl",然后使用 openpyxl 库中的方法来实现单元格颜色的添加。
下面是一个示例代码:
```python
import pandas as pd
from openpyxl import Workbook
from openpyxl.styles import PatternFill
# 创建 DataFrame
df = pd.DataFrame({'Name': ['Alice', 'Bob', 'Charlie'],
'Age': [25, 30, 35],
'Score': [80, 90, 85]})
# 创建 ExcelWriter 对象
writer = pd.ExcelWriter('sample.xlsx', engine='openpyxl')
# 将 DataFrame 写入 Excel 文件
df.to_excel(writer, sheet_name='Sheet1', index=False)
# 获取工作表对象
workbook = writer.book
worksheet = writer.sheets['Sheet1']
# 创建填充色对象
green_fill = PatternFill(start_color='00FF00', end_color='00FF00', fill_type='solid')
# 遍历单元格,添加填充色
for row in range(2, len(df)+2):
cell = worksheet.cell(row=row, column=3)
if cell.value >= 85:
cell.fill = green_fill
# 保存 Excel 文件
writer.save()
```
这段代码会将 DataFrame 保存到 sample.xlsx 文件中的 Sheet1 工作表中,并且会将分数列中大于等于 85 的单元格背景色设置为绿色。
阅读全文