Блог

Я строю линейный график в Altair (4.1.0) и хотел бы использовать прямую маркировку (аннотации) вместо обычной легенды.
Таким образом, текстовая метка для каждой строки (скажем, временных рядов) должна появляться только один раз и в самой правой точке оси x (в отличие от этого примера точечной диаграммы, обозначающего каждую точку данных).).
Хотя я могу использовать pandas для манипулирования данными для получения желаемых результатов, я думаю, что было бы более элегантно использовать реализацию pure-Altair, но, похоже, я не могу понять это правильно.

Например, учитывая следующие данные:

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

np.random.seed(10)
time = pd.date_range(start="10/21/2020", end="10/22/2020", periods=n)
data = pd.concat([
    pd.DataFrame({
        "time": time,
        "group": "One",
        "value": np.random.normal(10, 2, n)}),
    pd.DataFrame({
        "time": time,
        "group": "Two",
        "value": np.random.normal(5, 2, n)}).iloc[:-1]
], ignore_index=True)
  

Я могу сгенерировать удовлетворительный результат, используя pandas для создания подмножества, включающего последний момент времени для каждой группы:

 lines = alt.Chart(data).mark_line(
    point=True
).encode(
    x="time:T",
    y="value:Q",
    color=alt.Color("group:N", legend=None),  # Remove legend
)

text_data = data.loc[data.groupby('group')['time'].idxmax()]  # Subset the data for text positions
labels = alt.Chart(text_data).mark_text(
    # some adjustments
).encode(
    x="time:T",
    y="value:Q",
    color="group:N",
    text="group:N"
)

chart = lines   labels
  

Однако, если я попытаюсь использовать основные данные и добавить агрегации Altair, например, используя x=max(time) или явно transform_aggregate() , я либо получаю текстовые аннотации по всем точкам, либо вообще ничего (соответственно).

Есть ли лучший способ получить приведенный выше результат?

Вы можете сделать это, используя argmax агрегат в кодировке y. Например, ваш слой labels может выглядеть следующим образом:

 labels = alt.Chart(data).mark_text(
    align='left', dx=5
).encode(
    x='max(time):T',
    y=alt.Y('value:Q', aggregate={'argmax': 'time'}),
    text='group:N',
    color='group:N',
)