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

如何解决在 Altair 中使用不同颜色和标记绘制两个系列

我有一个如下所示的 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"],y="my_line_column:Q",color="variable:N"
).mark_line()

alt.layer(points,line)

该代码生成此图表：

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

解决方法

对于不那么冗长的解决方案，您可以使用其中一个图表作为“基础”，然后覆盖其变换、标记和 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