优草派  >   Python

pandas to_excel 添加颜色操作

孙悦            来源:优草派

Pandas 是 Python 中常用的数据分析库,它提供了一系列的数据处理和操作方法,其中 to_excel 方法可以将 DataFrame 数据写入到 Excel 表格中。在实际使用中,我们常常需要将表格中的数据按照一定的规则进行着色,以便更好地展示数据。本文将从以下几个角度来介绍如何在 Pandas 中使用 to_excel 添加颜色操作。一、使用 Style 方法添加颜色

Pandas 中的 Style 方法可以对 DataFrame 进行样式设置,包括单元格的背景色、字体颜色、对齐方式等。我们可以通过 Style 方法来实现对 Excel 表格的颜色设置。具体操作步骤如下:

pandas to_excel 添加颜色操作

1. 使用 Style 方法创建样式对象:

```python

import pandas as pd

import numpy as np

df = pd.DataFrame(np.random.randn(4,4), columns=list('ABCD'))

style = df.style

```

2. 使用 background_gradient 方法设置单元格背景色:

```python

style.background_gradient(cmap='RdYlGn')

```

其中 cmap 参数指定了颜色映射表,这里使用了红黄绿三种颜色进行渐变。运行结果如下:

![image-20211202210928210](https://i.loli.net/2021/12/02/1QKrBzv4cXhWZSj.png)

3. 使用 highlight_max 方法设置单元格最大值着色:

```python

style.highlight_max(color='lightgreen')

```

其中 color 参数指定了最大值的颜色,运行结果如下:

![image-20211202211119693](https://i.loli.net/2021/12/02/XVCS2M8q9H4GyJh.png)

二、使用 openpyxl 库添加颜色

除了使用 Pandas 的 Style 方法,我们还可以使用 openpyxl 库来实现对 Excel 表格的颜色设置。openpyxl 是一个处理 Excel 文件的 Python 库,它可以读取、修改和创建 Excel 文件。具体操作步骤如下:

1. 安装 openpyxl 库:

```python

pip install openpyxl

```

2. 导入库:

```python

import openpyxl

```

3. 使用 load_workbook 方法读取 Excel 文件:

```python

wb = openpyxl.load_workbook('test.xlsx')

sheet = wb.active

```

4. 使用 cell 方法获取单元格对象,使用 fill 方法设置单元格背景色:

```python

cell = sheet['A1']

cell.fill = openpyxl.styles.PatternFill(start_color='FFFF00', end_color='FFFF00', fill_type='solid')

```

其中 start_color 和 end_color 分别指定了起始和结束颜色,fill_type 指定了填充类型。运行结果如下:

![image-20211202211737908](https://i.loli.net/2021/12/02/vtGzX2E6JhKN8OU.png)

三、使用 xlsxwriter 库添加颜色

除了 openpyxl 库,我们还可以使用 xlsxwriter 库来实现对 Excel 表格的颜色设置。xlsxwriter 是一个用于创建 Excel 文件的 Python 库,它提供了灵活的样式设置方法。具体操作步骤如下:

1. 安装 xlsxwriter 库:

```python

pip install xlsxwriter

```

2. 导入库:

```python

import xlsxwriter

```

3. 创建 Workbook 对象:

```python

workbook = xlsxwriter.Workbook('test.xlsx')

worksheet = workbook.add_worksheet()

```

4. 使用 add_format 方法创建样式对象,使用 set_bg_color 方法设置单元格背景色:

```python

format1 = workbook.add_format({'bg_color': 'yellow'})

worksheet.write('A1', 'Hello', format1)

```

其中 bg_color 参数指定了背景色,运行结果如下:

![image-20211202212218337](https://i.loli.net/2021/12/02/6cvK4Su9w3HmL1R.png)

综上,我们介绍了在 Pandas 中使用 Style 方法、openpyxl 库和 xlsxwriter 库来实现 Excel 表格的颜色设置。通过调整背景色、字体颜色等,我们可以更好地展示数据,提高数据的可读性和可视化效果。

【原创声明】凡注明“来源:优草派”的文章,系本站原创,任何单位或个人未经本站书面授权不得转载、链接、转贴或以其他方式复制发表。否则,本站将依法追究其法律责任。
TOP 10
  • 周排行
  • 月排行