Можно ли добавить черные линии сетки на python графиков Altair Heatmap?

Question

Можно ли добавить линии Черной сетки в Altair Heatmap?

Я знаю, что можно добавить интервал между ячейками с scale.bandPaddingInner, и я подумал, что, возможно, можно было бы покрасить это интервал с черным?

То, что я хочу, выглядит примерно так: Тепловая карта с черными линиями сетки

Answer 1

Да, вы можете сделать это с помощью свойства stroke метки rect. Адаптация примера тепловой карты из документации Altair:

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