如何在Altair中用LabelExpression将数值轴刻度替换为文本

你的代码里labelExpr无法正常显示标签的原因有两个：

• y轴是离散类型（O），datum.y的实际取值是字符串形式的"0"、"10"、"20"，直接做减法运算会得到非预期结果；
• 就算转成数值，datum.y -1得到的是-1、9、19，完全超出了labels数组的索引范围（0、1、2），自然取不到对应的标签文本。

修正方案1：通过数值计算匹配数组索引 #

把datum.y转成数值后，除以10就能得到正好对应labels的索引（0→0，10→1，20→2），修改后的代码如下：

source = pd.DataFrame({
    'x': [0, 10, 20, 30, 0, 10, 20, 30, 0, 10, 20, 30],
    'y': [0, 0, 0, 0, 10, 10, 10, 10, 20, 20, 20, 20],
    'z': [4, 8, 1, 1, 0, 0, 0, 0, 0, 0, 0, 0]
})

labels = ['A', 'B', 'C']

alt.Chart(source).mark_rect().encode(
    x=alt.X('x:O'),
    y=alt.Y('y:O', axis=alt.Axis(
                labelExpr=f"['{labels[0]}', '{labels[1]}', '{labels[2]}'][parseInt(datum.y) / 10]",
                )
           ),
    color=alt.Color('z:Q')
)

修正方案2：使用条件判断映射标签 #

如果y轴的取值没有规律，也可以用条件表达式逐个匹配：

alt.Chart(source).mark_rect().encode(
    x=alt.X('x:O'),
    y=alt.Y('y:O', axis=alt.Axis(
                labelExpr="datum.y === '0' ? 'A' : datum.y === '10' ? 'B' : 'C'"
                )
           ),
    color=alt.Color('z:Q')
)

两种方案都能让y轴正确显示A、B、C标签，你可以根据实际数据的规律选择合适的方式。

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