python - Plotly:如何对两个不同图的类别使用相同的颜色？

Question

我正在尝试为类别创建两个具有相同配色方案的图。然而，这些图使用了不同的颜色模式，似乎我无法解决这个问题:

.

如何强制绘图对两个图表使用相同的颜色？我非常感谢任何帮助!

Image

Answer 1

定义离散颜色图。在下面的代码中: c = dict(zip(df["emotion"].unique(), px.colors.qualitative.G10))

import pandas as pd
import numpy as np
import plotly.express as px

# create some data
df = pd.DataFrame({"date":np.random.choice(pd.date_range("1-sep-2020","31-Dec-2020"),300),
             "emotion":np.random.choice(["positive","negative","anticipation","fear","trust"], 300)}).sort_values("date")

# map emotions to a color
c = dict(zip(df["emotion"].unique(), px.colors.qualitative.G10))

# bar chart
px.bar(
    df.groupby(["date", "emotion"], as_index=False)
    .size()
    .rename(columns={"size": "count"}),
    x="date",
    y="count",
    color="emotion",
    color_discrete_map=c
).show()

# pie chart
px.pie(
    df.groupby("emotion", as_index=False).agg(
        perc=("date", lambda s: len(s) / len(df))
    ),
    values="perc",
    names="emotion",
    color="emotion",
    color_discrete_map=c
).show()