如何在PyMC-Marketing生成的Matplotlib图中修改部分（而非全部）数据类别的标记样式

阿华AIGC实验室

2026-4-2

嘿，这个问题我刚好碰到过，给你分享几个实用的解决办法！因为PyMC-Marketing的绘图函数最终都是返回Matplotlib的Figure对象，我们完全可以直接拿到图里的元素手动调整，不用折腾底层源码。

核心思路

PyMC-Marketing生成的图本质上就是Matplotlib的Figure/Axes对象，我们只需要三步就能搞定：

拿到生成后的坐标轴（Axes）对象
定位到要修改的类别对应的线条/散点元素
单独给这些元素设置新的标记样式（比如三角形^、正方形s等）

具体操作步骤

1. 先获取生成的图和坐标轴

假设你最后一个单元格生成图的代码是类似这样的：

# 假设这是你生成图的代码（比如贡献图、ROI图等）
fig = mmm.plot_contributions()  # 或者其他PyMC-Marketing绘图函数

接下来我们从fig里拿到坐标轴对象：

# 如果是单张图，直接取第一个坐标轴
ax = fig.axes[0]
# 如果是多子图，根据你的需求选对应索引，比如fig.axes[1]是第二个子图

2. 确认要修改的类别对应关系

先搞清楚每个线条对应的是哪个类别，避免改错：

# 打印所有线条的标签，对应你的12个类别
for idx, line in enumerate(ax.lines):
    print(f"索引 {idx}: {line.get_label()}")

运行后你就能看到每个索引对应的类别名称，比如“抖音广告”“百度搜索”之类的，这样就能精准锁定要修改的目标。

3. 修改指定类别的标记样式

这里给你两种方式，选你觉得顺手的：

方式一：按索引修改

比如你想把第2、4、6个类别（索引从0开始，对应1、3、5）改成三角形标记：

# 定义要修改的索引和目标标记
target_indices = [1, 3, 5]
custom_marker = '^'  # 三角形，还可以选's'正方形、'D'菱形、'*'星号等

# 遍历线条，修改目标索引的样式
for idx, line in enumerate(ax.lines):
    if idx in target_indices:
        line.set_marker(custom_marker)
        line.set_markersize(8)  # 可选：调大标记，更显眼
        line.set_markeredgecolor('black')  # 可选：给标记加黑边，增强辨识度

方式二：按类别标签修改（更直观）

如果记不住索引，直接用类别名称筛选更省心：

custom_marker = '^'
# 比如你想把所有带"广告"的类别改成三角形
for line in ax.lines:
    if "广告" in line.get_label():
        line.set_marker(custom_marker)
        line.set_markersize(8)

4. 刷新图表生效

修改完后，刷新一下图就能看到效果了：

fig.canvas.draw()

特殊情况：如果是散点图怎么办？

如果你的图是散点类型（比如某些分布散点图），元素会存在ax.collections里，处理逻辑差不多：

# 确认散点的标签（如果有的话）
for idx, scatter in enumerate(ax.collections):
    print(f"索引 {idx}: {scatter.get_label()}")

# 修改指定散点的标记
target_indices = [2, 4]
for idx, scatter in enumerate(ax.collections):
    if idx in target_indices:
        scatter.set_marker('^')
        scatter.set_sizes([100])  # 调整散点大小