Altair热图在文本上打勾

时间:2020-09-28 14:33:32

标签: python altair

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

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

这是一个最小的示例,其中我将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'
)

enter image description here

1 个答案:

答案 0 :(得分:2)

执行此操作的一种方法是使用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'
)

enter image description here