matplotlib - Matplotlib中带有圆角的条形图？

Question

如何创建带有圆角的条形图，如该图所示？可以用matplotlib完成吗？

Answer 1

似乎无法直接在条形图中添加圆角。但是matplotlib确实提供了 FancyBboxPatch 类，该示例的演示版可用here。

因此，为了创建一个如问题所示的图表，我们首先可以制作一个简单的水平条形图:

import pandas as pd
import numpy as np
# make up some example data
np.random.seed(0)
df = pd.DataFrame(np.random.uniform(0,20, size=(4,4)))
df = df.div(df.sum(1), axis=0)
# plot a stacked horizontal bar chart
ax = df.plot.barh(stacked=True, width=0.98, legend=False)
ax.figure.set_size_inches(6,6)

这将产生以下图:



为了使矩形具有圆角，我们可以遍历 ax.patches中的每个矩形补丁并将其替换为 FancyBboxPatch。这个新的带有圆角的精美补丁可以复制旧补丁中的位置和颜色，因此我们不必担心放置。

from matplotlib.patches import FancyBboxPatch
ax = df.plot.barh(stacked=True, width=1, legend=False)
ax.figure.set_size_inches(6,6)
new_patches = []
for patch in reversed(ax.patches):
    bb = patch.get_bbox()
    color=patch.get_facecolor()
    p_bbox = FancyBboxPatch((bb.xmin, bb.ymin),
                        abs(bb.width), abs(bb.height),
                        boxstyle="round,pad=-0.0040,rounding_size=0.015",
                        ec="none", fc=color,
                        mutation_aspect=4
                        )
    patch.remove()
    new_patches.append(p_bbox)
for patch in new_patches:
    ax.add_patch(patch)

这就是我们得到的:



我给盒子加上了负填充，以使条形之间留有空隙。这些数字有点黑魔法。不知道 rounding_size和 pad的单位是什么。上一个演示示例中显示了 mutation_aspect，此处将其设置为4，因为 y范围约为4，而 x范围约为1。