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

在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

Answer 2

有两种方法可以完成您想要的事情：

使用pandas过滤数据（对我来说，这是最简单但并不总是简单或可用的具有过滤器的vega-lite版本：transform_window用于对数据进行排名，而transform_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-lite版本：

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)

这两个代码均产生以下图表：

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

2 个答案: