首页 > 解决方案 > Altair - 用不同颜色表示负和正的区域图

问题描述

我的 pandas DataFrames 中有一个包含正值和负值的列,我需要为正负 y 轴制作一个具有不同颜色的区域图。

到目前为止,我无法使用 alt.condition 做到这一点

brush = alt.selection(type='interval', encodings=['x'])

upper = alt.Chart(yData['plotY'].fillna(0).reset_index()[24000:26000],
                  title = '_name').mark_area().encode(x = alt.X('{0}:T'.format(yData['plotY'].index.name),
                                                                scale = alt.Scale(domain=brush)),
                                                      y = 'plotY',
#                                                       color=alt.condition(
#                                                             alt.datum.plotY > 0,
#                                                             alt.value("steelblue"),  # The positive color
#                                                             alt.value("orange")  # The negative color
#                                                         ),
                                                      tooltip=['plotY']).properties(width = 700,
                                                                                  height = 230)

lower = upper.copy().properties(
    height=20
).add_selection(brush)

p = alt.vconcat(upper, lower).configure_concat(spacing=0)
p

图形

如何制作具有不同颜色的正负图?

标签: pythonvisualizationaltair

解决方案


你可以这样做:

import altair as alt
import pandas as pd
import numpy as np

x = np.linspace(0, 100, 1000)
y = np.sin(x)
df = pd.DataFrame({'x': x, 'y': y})

alt.Chart(df).transform_calculate(
    negative='datum.y < 0'
).mark_area().encode(
    x='x',
    y=alt.Y('y', impute={'value': 0}),
    color='negative:N'
)

在此处输入图像描述

一些注意事项:

  • 我们使用计算的颜色编码而不是颜色条件,因为编码实际上会将数据分成两组,这是区域标记所必需的(区域标记与点标记不同,为每组数据绘制单个图表元素,以及单个图表元素不能有多种颜色)

  • to的impute参数y很重要,因为它告诉每个组在未定义且已定义另一组的情况下将值视为零。这可以防止在组中的点之间绘制直线的奇怪伪影。


推荐阅读