如何在 Altair 条形图中突出显示具有不同 colors 的 2 个条形图？

Question

我想在 Altair 条形图中突出显示两个不同的条形。 但是，根据不同的条件，每个人都必须具有不同的颜色。

基础知识：

import altair as alt
import pandas as pd

df = pd.DataFrame({"fruits": ["banana", "plum", "apple", "lime"], "values": [10, 6, 8, 5]})

plot = (
    alt.Chart(df)
    .mark_bar()
    .encode(
        x=alt.X("fruits", sort="y"),
        y="values"))

apple_plot = plot.encode(color=alt.condition(alt.datum.fruits == "apple", alt.value("red"), alt.value("navy")))

banana_plot = plot.encode(color=alt.condition(alt.datum.fruits == "banana", alt.value("yellow"), alt.value("navy")))

apple_plot | banana_plot

到目前为止的结果

我试过的

apple_banana_plot = plot.encode(
    color=alt.condition(
        alt.datum.fruits == "apple",
        alt.value("red"),
        alt.condition(
            alt.datum.fruits == "banana",
            alt.value("yellow"),
            alt.value("navy"))))

apple_banana_plot

我得到的错误：

TypeError：类型 object 获得了关键字参数“条件”的多个值

...这意味着我不能在另一个条件中使用条件。

我期待的结果

Answer 1

你可以这样做：基本上，你用两个参数定义color ：

• domain - 分类列中的值列表
• 范围 - 具体值 colors

df = pd.DataFrame({"fruits": ["banana", "plum", "apple", "lime"], "values": [10, 6, 8, 5]})

plot = (
    alt.Chart(df)
    .mark_bar()
    .encode(
        x=alt.X("fruits", sort="y"),
        y="values",
    color=alt.Color('fruits',
                     scale=alt.Scale(
                         domain=df.sort_values(['values'])['fruits'].tolist(),
                         range=['navy', 'navy', 'red', 'yellow']))))
plot