Python - 使用 pandas 格式化 Excel 单元格

在 Python 中，使用 pandas库来格式化 Excel 单元格通常涉及以下步骤：

1. 导入 pandas 和 openpyxl 库

openpyxl 是处理 Excel 文件的库，它支持 Excel 2010 及以上版本的 xlsx/xlsm/xltx/xltm 格式。

import pandas as pdfrom openpyxl import load_workbook

2. 读取 Excel 文件

使用 pandas 的 read_excel() 函数来加载 Excel 文件到 DataFrame 中。

df = pd.read_excel('file.xlsx')

3. 格式化单元格

3.1 设置单元格样式

设置字体、颜色等样式。

添加条件格式（例如，根据数值大小改变背景色）。

4. 保存修改后的 Excel 文件

使用 to_excel() 函数将 DataFrame 写入到新的或已存在的 Excel 文件中。

writer = pd.ExcelWriter('output.xlsx', engine='openpyxl')df.to_excel(writer, index=False)writer.save()

5. 应用条件格式示例

假设我们有一个列 'age'，我们想根据年龄的不同设置不同的背景色。

def set_background_color(row):    if row['age'] < 18:        return 'lightcoral'  # 青少年背景颜色    elif row['age'] >= 18 and row['age'] < 35:        return 'lightskyblue'  # 青年背景颜色    else:        return 'lavender'  # 成年背景颜色

df['bg_color'] = df.apply(set_background_color, axis=1)

6. 测试用例

import pandas as pdtest_df = pd.DataFrame({    'name': ['Alice', 'Bob', 'Charlie'],    'age': [20, 40, 30]})def test_set_background_color():    assert set_background_color(test_df.iloc[0]) == 'lightskyblue'    assert set_background_color(test_df.iloc[1]) == 'lavender'    assert set_background_color(test_df.iloc[2]) == 'lightcoral'test_set_background_color()

7. 应用场景和示例

假设我们要使用 AI 来辅助数据可视化。比如，我们可以通过 AI 自动分析一个 Excel 文件，确定哪些列应该以哪种颜色或样式显示，然后根据这些规则更新 Excel 文件的格式。

应用场景

数据分析报告的可视化

示例

AI 可以分析数据，如果某个年龄段的数据很多（例如 30 岁以上），那么对应的背景色可能会更加显著，从而方便用户快速识别。

转载地址：http://liafk.baihongyu.com/

你可能感兴趣的文章