如何在 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