在 Altair 中使用不同的颜色和标记绘制两个系列答案

【问题标题】：Plotting two series using different colors and marks in Altair在 Altair 中使用不同的颜色和标记绘制两个系列
【发布时间】：2021-07-18 02:14:24
【问题描述】：

我有一个看起来像这样的 Pandas 数据框：

| date_column         |   my_point_column |   my_line_column |
|---------------------|-------------------|------------------|
| 2017-12-13 00:00:00 |                 3 |          9.73136 |
| 2018-01-05 00:00:00 |                23 |          6.36568 |
| 2018-01-11 00:00:00 |                 6 |          11.9105 |
| 2018-01-22 00:00:00 |                11 |          10.4328 |
| ...                 |               ... |              ... |

我想用不同的标记（点和线）绘制这两个系列，它们应该是不同的颜色。颜色也必须显示在图例中。

我可以用这段代码完成它，但它非常冗长：

# plot_df is the dataframe shown above
chart = alt.Chart(plot_df)

points = chart.transform_fold(
    fold=["my_point_column"],
    as_=["variable", "value"]
).encode(
    x="date_column:T",
    y="my_point_column:Q",
    color="variable:N"
).mark_circle()

line = chart.transform_fold(
    fold=["my_line_column"],
    as_=["variable", "value"]
).encode(
    x="date_column:T",
    y="my_line_column:Q",
    color="variable:N"
).mark_line()

alt.layer(points, line)

该代码生成此图表：

这几乎就是我正在寻找的结果（除了自定义颜色、轴标题等）。有没有更简单的方法来做到这一点而不必使用 transform_fold 两次？

【问题讨论】：

标签： python altair

【解决方案1】：

对于一个不太冗长的解决方案，您可以使用其中一个图表作为“基础”，然后覆盖其转换、标记和 y 编码：

points = alt.Chart(plot_df).mark_circle().transform_fold(
    fold=["my_point_column"]
).encode(
    x="date_column:T",
    y="my_point_column:Q",
    color="key:N")

line = chart.mark_line().transform_fold(
    fold=["my_line_column"]
).encode(
    y="my_line_column:Q")

points + line

【讨论】：