plotnine中如何更优地在after_stat中引用facet_wrap参数？

你不需要将分面变量传入aes来计算百分比，利用plotnine的after_stat结合窗口函数语法，就能在分面后的每个子图内直接计算各组的占比，完全避免手动抵消aes设置的操作。

核心思路是：

• facet_wrap会自动将数据按vs拆分到各个子图中，每个子图内的统计计算是独立的。
• 在after_stat中使用sum(count) over (partition by x)，可以在每个子图内按x轴变量（cyl）分组，计算该组的总计数，再用单个类别的count除以这个总和得到占比。

修改后的代码 #

from plotnine import *
from plotnine.data import mtcars

(
    ggplot(mtcars, aes('factor(cyl)', fill='factor(am)'))
    + geom_bar(position='fill')
    + geom_text(
        aes(label=after_stat('(count / sum(count) over (partition by x)) * 100')),
        stat='count',
        position=position_fill(vjust=0.5),
        format_string='{:.1f}%',
        show_legend=False
    )
    + facet_wrap('vs')
).draw()

为什么这个方法更优 #

1. 无需额外aes设置：不再需要把分面变量vs塞进color aes，也不用手动设置scale_color_manual来抵消，代码更简洁。
2. 统计逻辑更清晰：直接利用分面后的数据上下文，窗口函数自动在每个子图内完成分组计算，符合可视化的逻辑流程。
3. 性能更高效：避免了额外的数据转换（比如你原代码中创建临时DataFrame的操作），直接在统计阶段完成计算。

如果需要更复杂的分组逻辑，也可以自定义统计变换，但对于这个场景，窗口函数的方法已经足够简洁高效。

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