python - 使用python / plotly express在条形图和饼图中使用相同的颜色

Question

我对此感到困惑：我尝试用条形图每月表示相同的数据，用饼图表示总数，但我无法在两个图表上获得表示同一类别的相同颜色。我的代码如下所示：

import pandas as pd
import plotly.express as px
from plotly.offline import plot

df = pd.read_csv('myfile.csv')
data = df.groupby(['month', 'category']).size().reset_index(name='number of records')
histo = px.bar(data, x='month', y='number of records', color='category', barmode='stack')
pie = px.pie(data, values='number of records', names='category')
plot(histo, filename = 'histo')
plot(pie, filename = 'pie')

结果是：

我们可以看到 A 在条形图上是蓝色的，而 X 在饼图上是蓝色的。

任何人都可以帮忙吗？

Tks

标签： pythoncolorsplotlyplotly-python

Answer 1

好的，最后很明显：只需在饼图中添加颜色键，如下所示：

pie = px.pie(data, values='number of records', names='category', color='category')