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

Question

我有一个 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 两次？

Answer 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

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

Plotting two series using different colors and marks in Altair

python

altair