Altair 热图刻度文本

Altair heatmap ticks text

有没有办法将刻度文本设置为字符串列表?我知道理想情况下最好直接在数据集中使用刻度,但不幸的是,这并不总是可行或实用。

我正在制作热图,我希望能够设置要显示的刻度值,然后在这些位置放置特定的字符串。

这是一个最小的示例,其中我在 x 轴上将刻度值设置为 2pi。

# Compute x^2 + y^2 across a 2D grid
x, y = np.meshgrid([i*np.pi for i in range(10)], [i*np.pi for i in range(10)])
z = x ** 2 + y ** 2

# Convert this grid to columnar data expected by Altair
source = pd.DataFrame({'x': x.ravel(),
                     'y': y.ravel(),
                     'z': z.ravel()})

alt.Chart(source).mark_rect().encode(
    alt.X('x:O', axis=alt.Axis(values=[2*i*np.pi for i in range(5)], format=".2f")),
    alt.Y('y:O', axis=alt.Axis(format=".2f")),
    color='z:Q'
)

一种方法是使用 labelExpr 轴属性,它允许使用 Vega Expression 语法定义新的标签值。例如:

alt.Chart(source).mark_rect().encode(
    alt.X('x:O',
      axis=alt.Axis(
        values=[2*i*np.pi for i in range(5)],
        labelExpr="round(datum.value / PI) + 'π'"
      )
    ),
    alt.Y('y:O',
      axis=alt.Axis(
        labelExpr="round(datum.value / PI) + 'π'"
      )
    ),
    color='z:Q'
)