如何全局修改plotnine默认配色方案与X轴日期刻度格式？

一、全局设置自定义配色方案 #

plotnine没有直接的scale_set()函数，但可以通过两种实用方式实现全局配色：

方法1：修改scale的默认参数 #

直接覆盖对应scale的default属性，针对离散型颜色的示例：

from plotnine import scales, ggplot, aes, geom_point, theme_set, theme_minimal
import pandas as pd
import numpy as np

# 定义自定义配色板
custom_palette = ["#FF5733", "#33FF57", "#3357FF", "#FF33A6"]

# 覆盖离散颜色、填充scale的默认值
scales.scale_color_discrete.defaults['palette'] = lambda n: custom_palette[:n]
scales.scale_fill_discrete.defaults['palette'] = lambda n: custom_palette[:n]

# 测试：无需手动添加scale_color_manual
df = pd.DataFrame({
    'x': np.random.rand(50),
    'y': np.random.rand(50),
    'group': np.repeat(['A','B','C','D'], 12)[:50]
})
(ggplot(df, aes(x='x', y='y', color='group'))
 + geom_point(size=3)
 + theme_minimal())

连续型配色同理，修改scale_color_continuous.defaults的cmap或渐变函数参数即可。

方法2：绑定scale到全局主题 #

把自定义scale和主题整合后，通过theme_set()全局生效：

custom_theme = theme_minimal() + scales.scale_color_manual(values=custom_palette) + scales.scale_fill_manual(values=custom_palette)
theme_set(custom_theme)

# 后续绘图自动应用配色
(ggplot(df, aes(x='x', y='y', color='group')) + geom_point(size=3))

二、全局设置日期时间轴刻度格式 #

直接修改scale_x_datetime的默认参数，实现全局统一配置：

from plotnine import scales
from plotnine.data import economics

# 覆盖x轴日期刻度的默认breaks和labels
scales.scale_x_datetime.defaults['breaks'] = scales.date_breaks('5 years')
scales.scale_x_datetime.defaults['labels'] = scales.date_format('%Y')

# 测试：无需重复设置轴格式
(ggplot(economics, aes(x='date', y='unemploy'))
 + geom_line()
 + theme_minimal())

需要修改y轴日期格式的话，同理调整scales.scale_y_datetime.defaults的对应参数。

三、类Matplotlib风格的一键全局配置 #

把所有全局设置封装成函数，脚本开头调用一次即可完成所有配置：

def set_plotnine_global_style():
    # 设置基础主题
    theme_set(theme_minimal())
    
    # 配置自定义配色
    custom_palette = ["#FF5733", "#33FF57", "#3357FF", "#FF33A6"]
    scales.scale_color_discrete.defaults['palette'] = lambda n: custom_palette[:n]
    scales.scale_fill_discrete.defaults['palette'] = lambda n: custom_palette[:n]
    
    # 配置日期轴格式
    scales.scale_x_datetime.defaults['breaks'] = scales.date_breaks('5 years')
    scales.scale_x_datetime.defaults['labels'] = scales.date_format('%Y')

# 初始化全局样式
set_plotnine_global_style()

# 后续所有绘图自动应用配置
(ggplot(economics, aes(x='date', y='unemploy', color='pop')) + geom_line())

内容的提问来源于stack exchange，提问作者dsforecast