在Python中更改堆积条形图图例

Question

我在csv文件中有以下数据：

Date    City    TruckA  TruckB  TruckC  TruckD
Date1   City1   1   0   0   0
Date1   City2   0   0   1   0
Date1   City3   1   0   0   0
Date1   City4   0   0   1   0
Date2   City1   1   0   0   0
Date2   City2   0   1   0   0
Date2   City3   0   0   0   1
Date2   City4   1   0   0   0
Date2   City5   0   1   0   0
Date3   City1   1   0   0   0
Date3   City2   0   0   1   0
Date3   City3   1   0   0   0
Date3   City4   0   0   1   0

我可以用以下代码成功绘制数据：

^{pr2}$

我得到以下结果：

正如你所看到的，颜色的传奇是这样的每一对（城市，卡车）有一个颜色。我希望传奇只依赖于卡车，理想的是每个城市的条形图上都有标签。在

这可能吗？在

Answer 1

编辑

import matplotlib.pyplot as plt
cycle = plt.rcParams['axes.prop_cycle'].by_key()['color']
df_out = df.unstack()
d = dict(zip(df.columns.get_level_values(0),cycle))
c = df_out.columns.get_level_values(0).map(d)
df_out.plot.bar(stacked=True, color=c, figsize=(10,8))

输出：

添加了edgecolor以区分城市：

^{pr2}$

---

IIUC，我想你在找这样的东西：

df = df.set_index(["Date","City"])
df.sum(level=0).plot.bar(stacked=True, figsize=(10,8))

输出：