使用matplotlib中的dataframe.plot（）函数编辑条的宽度

Question

我正在使用以下方式制作堆积条形图：

DataFrame.plot(kind='bar',stacked=True)

我想控制条的宽度，以便条形图像直方图一样相互连接。

我查看了文档，但无济于事 - 有什么建议吗？是否可以这样做？

Answer 1

对于遇到此问题的任何人：

由于pandas 0.14，使用条形图绘制时有一个'width'命令： https://github.com/pydata/pandas/pull/6644

现在只需使用

即可解决上述示例

df.plot(kind='bar', stacked=True, width=1)

Answer 2

如果认为你必须使用matplotlib“后处理”条形图，因为pandas在内部设置条形的宽度。

形成条形的矩形位于容器对象中。 因此，您必须遍历这些容器并单独设置矩形的宽度：

In [208]: df = pd.DataFrame(np.random.random((6, 5)) * 10,               
                        index=list('abcdef'), columns=list('ABCDE'))

In [209]: df
Out[209]: 
     A    B    C    D    E
a  4.2  6.7  1.0  7.1  1.4
b  1.3  9.5  5.1  7.3  5.6
c  8.9  5.0  5.0  6.7  3.8
d  5.5  0.5  2.4  8.4  6.4
e  0.3  1.4  4.8  1.7  9.3
f  3.3  0.2  6.9  8.0  6.1

In [210]: ax = df.plot(kind='bar', stacked=True, align='center')

In [211]: for container in ax.containers:
              plt.setp(container, width=1)
   .....:         

In [212]: x0, x1 = ax.get_xlim()

In [213]: ax.set_xlim(x0 -0.5, x1 + 0.25)
Out[213]: (-0.5, 6.5)

In [214]: plt.tight_layout()

Answer 3

“我想控制条形的宽度，使条形像直方图一样相互连接。”

更好的选择是使用 sns.displot()

示例代码：

emp = pd.read_csv("https://raw.githubusercontent.com/arora123/Data/master/emp-data.csv")

sns.displot(emp, x='Department', hue='Gender', multiple='stack',  
             height=8, aspect=1.7);

Answer 4

matplotlib 解决方案

根据需要修改 width 中的 ax.bar 参数

代码

import numpy as np
import pandas as pd
from matplotlib import pyplot as plt

df = pd.DataFrame(np.arange(15).reshape(5, 3), columns=list('ABC'))

print(df)

fig, axs = plt.subplots(1, 2)

ax = axs[0]
xs = np.arange(df.shape[1])
ys = np.zeros(xs.shape)
for ind in df.index:
    ax.bar(xs, df.loc[ind, :], label=ind, bottom=ys, width=.4)
    ys += df.loc[ind, :]
plt.setp(ax, xticks=xs, xticklabels=list(df))
ax.legend(title='rows')
ax.set_xlabel('columns')

ax = axs[1]
xs = np.arange(df.shape[0])
ys = np.zeros(xs.shape)
for col in list(df):
    ax.bar(xs, df.loc[:, col], label=col, bottom=ys, width=.4)
    ys += df.loc[:, col]
plt.setp(ax, xticks=xs, xticklabels=df.index.to_numpy().tolist())
ax.legend(title='columns')
ax.set_xlabel('rows')

plt.show()

df=

    A   B   C
0   0   1   2
1   3   4   5
2   6   7   8
3   9  10  11
4  12  13  14