python - Altair 热图标记文本

Question

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

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

这是一个最小的示例，我在 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'
)

Answer 1

一种方法是使用labelExpr轴属性，它允许使用Vega 表达式语法定义新的标签值。例如：

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'
)