python - 是否可以将黑色网格线添加到 python Altair Heatmap 图中？

Question

是否可以将黑色网格线添加到 Altair 热图？

我知道可以使用 scale.bandPaddingInner 在单元格之间添加间距，我想也许可以用黑色为该间距着色？

我想要的是这样的:Heatmap with Black Grid lines

Answer 1

是的，您可以使用 stroke 标记的 rect 属性执行此操作。改编 Altair 文档中的 heatmap example:

import altair as alt
import numpy as np
import pandas as pd

# Compute x^2 + y^2 across a 2D grid
x, y = np.meshgrid(range(-5, 5), range(-5, 5))
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(stroke='black', strokeWidth=2).encode(
    x='x:O',
    y='y:O',
    color='z:Q'
)