如何在含transform fold的Altair堆叠条形图中自定义堆叠顺序？

问题描述 #

使用宽格式数据时，需通过transform_fold转换为长格式才能生成堆叠条形图，但默认堆叠顺序为字母序。尝试过修改transform_fold列顺序、设置X轴sort参数、添加order字段、设置color的sort参数，均无法改变堆叠顺序，求可行的自定义方法。

测试代码 #

import pandas as pd
import altair as alt

small_df = pd.DataFrame({
    'Probability A':[0.2,0.8,0.4],
    'Probability B':[0.5,0.1,0.1],
    'Probability C':[0.3,0.1,0.5],
    'ID':['John','Sally','Frank']
})

alt.Chart(small_df).mark_bar().transform_fold(
        ['Probability A','Probability C','Probability B']
    ).encode(
        x=alt.X('value:Q',scale=alt.Scale(domain=[0, 1]),sort=['Probability A','Probability C','Probability B']),
        y='ID:N',
        order=['Probability A','Probability C','Probability B'],
        color=alt.Color('key:N',sort=['Probability A','Probability C','Probability B'],scale=alt.Scale(domain=['Probability A','Probability B','Probability C'],range=['red', 'orange', 'green']))
)

当前效果 #

解决方案 #

要自定义堆叠顺序，核心是给每个折叠后的key分配数值排序权重，再通过order编码绑定这个权重字段。具体实现如下：

1. 在transform_fold后添加transform_calculate，为每个目标key映射对应的排序优先级（数值越小，堆叠时越靠前）；
2. 将order编码设置为新生成的排序字段；
3. 若需同步图例顺序，保持color的sort参数与自定义顺序一致即可。

修改后的完整代码：

import pandas as pd
import altair as alt

small_df = pd.DataFrame({
    'Probability A':[0.2,0.8,0.4],
    'Probability B':[0.5,0.1,0.1],
    'Probability C':[0.3,0.1,0.5],
    'ID':['John','Sally','Frank']
})

custom_order = ['Probability A', 'Probability C', 'Probability B']

alt.Chart(small_df).mark_bar().transform_fold(
        custom_order
    ).transform_calculate(
        # 按自定义顺序给每个key分配排序值
        sort_order=alt.expr.indexof(alt.expr.array(*custom_order), alt.datum.key)
    ).encode(
        x=alt.X('value:Q', scale=alt.Scale(domain=[0, 1])),
        y='ID:N',
        # 用排序值控制堆叠顺序
        order='sort_order:Q',
        color=alt.Color(
            'key:N',
            sort=custom_order,  # 同步图例顺序
            scale=alt.Scale(domain=custom_order, range=['red', 'green', 'orange'])
        )
)

关键说明 #

之前的尝试无效原因：

• X轴的sort参数控制的是轴刻度顺序，而非堆叠内部的条形顺序；
• order字段直接传入字符串列表不生效，必须绑定可用于排序的数值字段；
• color的sort仅控制图例颜色顺序，不会影响堆叠层级顺序。

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