pandas给单元格变颜色|极客笔记

pandas给单元格变颜色

在数据分析中，经常使用pandas库来处理和分析数据。pandas库提供了丰富的功能和方法，用于数据的处理和可视化。有时候我们需要对数据进行高亮标记或者显示特殊的样式，这时可以使用pandas的样式功能来实现单元格变色。

1. 基本概念

在pandas中，可以使用Styler对象来设置单元格的样式。Styler对象有一个applymap方法，可以接受一个函数作为参数，该函数根据单元格的值返回样式字符串。通过applymap方法可以为不同的单元格设置不同的样式。

2. 示例

下面我们通过一个示例来演示如何使用pandas给单元格变颜色。

import pandas as pd

# 创建一个DataFrame
data = {'Name': ['Alice', 'Bob', 'Cathy', 'David'],
        'Score': [85, 90, 88, 92]}

df = pd.DataFrame(data)

def color_high_score(val):
    color = 'red' if val > 90 else 'black'
    return 'color: %s' % color

# 使用Styler对象的applymap方法设置单元格颜色
styled = df.style.applymap(color_high_score, subset=['Score'])

styled

运行以上代码，可以看到输出的DataFrame中分数大于90的单元格会显示为红色，小于等于90的单元格显示为黑色。

3. 设置多个样式

除了设置单一的样式之外，我们还可以设置多个样式来丰富数据的显示效果。

import pandas as pd

# 创建一个DataFrame
data = {'Name': ['Alice', 'Bob', 'Cathy', 'David'],
        'Math': [85, 90, 88, 92],
        'English': [75, 82, 80, 88]}

df = pd.DataFrame(data)

def color_high_score(val):
    color = 'red' if val > 90 else 'black'
    return 'color: %s' % color

def highlight_max(s):
    is_max = s == s.max()
    return ['color: red' if v else '' for v in is_max]

# 使用Styler对象的applymap方法设置单元格颜色
styled = df.style.applymap(color_high_score, subset=['Math']) \
                   .apply(highlight_max, subset=['Math', 'English'])

styled

以上代码中，我们定义了两个样式函数color_high_score和highlight_max，分别用来设置分数大于90的单元格为红色和标记出每一列的最大值。在apply方法中设置多个样式函数，可以为不同的列设置不同的样式。