如何将单个文本标签添加到 Altair 散点图中的最大点？

Question

我正在尝试在 Altair 中创建散点图，并且我想向具有最大 y 的点添加文本标签或注释。

我已经能够将 text_marks 添加到所有点，但我无法弄清楚如何根据 max(y) 标记特定点

以下是一些示例数据：

df = pd.DataFrame(columns=['date', 'daily', 'total'], 
         data=[['2019-08-01', 29, 102370],
               ['2019-08-02', 18, 102388],
               ['2019-08-03', 19, 102407],
               ['2019-08-04', 13, 102420],
               ['2019-08-05', 29, 102449],
               ['2019-08-06', 49, 102498],
               ['2019-08-07', 31, 102529],
               ['2019-08-08', 39, 102568],
               ['2019-08-09', 23, 102591],
               ['2019-08-10', 17, 102608],
               ['2019-08-11', 18, 102626],
               ['2019-08-12', 38, 102664],
               ['2019-08-13', 22, 102686]])

这是我到目前为止想出的，但它没有做我想要的，我想我可能把它复杂化了

chart = alt.Chart(
  data=df,
).mark_line(
    color='red'
).encode(
    alt.X('date:T', title=''),
    alt.Y('daily:Q', title='')
)

text = alt.Chart(df).mark_text().encode(
    x=alt.X('max(date):T'),
    y=alt.Y('max(daily):Q'),
    text=alt.Text('max(daily):Q')
)


(chart + text)

Answer 1

有两种方法可以做你想做的事：

1. 过滤数据pandas（对我来说这是最简单的但并不总是简单或可用的，
2. vega-lite 版本，带有过滤器：atransform_window对数据进行排名，atransform_filter仅保留最大值

熊猫版：

chart = (
    alt.Chart(data=df)
    .mark_line(color="red")
    .encode(alt.X("date:T", title=""), alt.Y("daily:Q", title=""))
)

text = (
    alt.Chart(df.query("daily == daily.max()"))
    .mark_text(dy=-15, color="red")
    .encode(x=alt.X("date:T"), y=alt.Y("daily:Q"), text=alt.Text("daily:Q"))
)


(chart + text)

Vega精简版：

chart = (
    alt.Chart(data=df)
    .mark_line(color="red")
    .encode(alt.X("date:T", title=""), alt.Y("daily:Q", title=""))
)

text = (
    alt.Chart(df)
    .mark_text(dy=-15, color="red")
    .transform_window(
        sort=[alt.SortField("daily", order="descending")], 
        rank="rank(daily)"
    )
    .transform_filter(alt.datum.rank == 1)
    .encode(x=alt.X("date:T"), 
            y=alt.Y("daily:Q"), 
            text=alt.Text("daily:Q"))
)


(chart + text)

两个代码都生成了下面的图表：

Answer 2

在 Vega-Lite 语法中执行此操作的一种简洁方法是使用argmax聚合。例如：

chart = alt.Chart(
  data=df,
).mark_line(
    color='red'
).encode(
    alt.X('date:T', title=''),
    alt.Y('daily:Q', title='')
)

text = alt.Chart(df).mark_text(dy=-15, color="red").encode(
    x=alt.X('date:T', aggregate={'argmax': 'daily'}),
    y=alt.Y('max(daily):Q'),
    text=alt.Text('max(daily):Q')
)

chart + text