使用Panda和Matplotlib绘制2个堆叠系列

Question

我正拼命想用Matplot创建漂亮的图形，但这并非易事。要进行语境化，我有两个系列（ serie1 ， serie2 ）。对于每个

我有3个组（Group1，Group2和Group3）。对于每个小组，我都有一些主题和价值观。每个系列通过不同的变量（主题）描述几个人（G1，G2，G3）的行为。代码是：

import pandas as pd
d = {"ThemeA": [25,34,75], "ThemeB": [0,71,18], "ThemeC": [2,0,0], "ThemeD":[1,14,0] }
serie1 = pd.DataFrame(data = d, index=["Groupe 1", "Groupe 2", "Groupe 3"] )
serie1= serie1.loc[:,:].div(serie1.sum(1), axis=0) * 100

d = {"ThemeA": [145,10,3], "ThemeB": [10,1,70], "ThemeC": [34,1,2], "ThemeD":[3,17,27]}
serie2= pd.DataFrame(data = d, index=["Groupe 1", "Groupe 2", "Groupe 3"])
serie2= serie2.loc[:,:].div(serie2.sum(1), axis=0) * 100

现在我想制作一个显示用户数据的图表：

ax = fig.add_subplot(111) 
ax = serie1.plot(kind='barh', ax=ax, width=0.2, stacked=True, position=0, sharex=True, 
             sharey=True, legend=True, figsize = (6,2))

serie2.plot(kind='barh', ax=ax, width=0.2, stacked=True, position=1.6, 
               sharex=True, sharey=True, legend=False)
ax.grid(False)
plt.ylim([-0.5, 2.5])

我能够得到以下图表：

但我想把传说移到底部。如果我尝试这样做，

ax.legend(loc='upper center', bbox_to_anchor=(0.5, -0.05), 
           fancybox=True, shadow=True, ncol=5)

我得到以下输出，标签太多了。

当然，我想在传奇中看到每个标签一次 如果有人有奇迹解决方案，我就是接受者！提前谢谢。

Answer 1

您可以使用比需要更长的x轴来为图例添加空白

# calculate the size of the longer column (max of row sums)
max_col = serie2.sum(axis=1).max()
# increase the size of the x axis a factor of 1.4
xlim(0, max_col*1.4)

如果您希望图例位于底部，当您致电legend时，您实际上是从两个图中绘制标签。您需要删除重复的标签。为此你使用字典。

from collections import OrderedDict

fig = figure()
figsize(6,2)
ax = fig.add_subplot(111) 

serie1.plot(kind='barh', ax=ax, width=0.2, stacked=True, position=0,
            sharex=True, sharey=True)

serie2.plot(kind='barh', ax=ax, width=0.2, stacked=True, position=1.6, 
            sharex=True, sharey=True)

handles, labels = gca().get_legend_handles_labels()
my_labels = OrderedDict(zip(labels, handles))
legend(my_labels.values(), my_labels.keys(), loc='upper center',
       bbox_to_anchor=(0.5, -0.1), fancybox=True, shadow=True, ncol=5)

ax.grid(False)
ylim([-0.5, 2.5])

然后你得到：

Answer 2

在这种情况下工作的单行黑客是添加行

serie2.columns= ["_" + col for col in serie2.columns]

在绘制第二个数据帧之前

。这将使用下划线替换所有列名称，后跟原始名称。由于以下划线（"_"）开头的名称未显示在图例中，因此只留下第一个数据框的图例条目。
此解决方案要求在两个数据帧中具有相同的列顺序。