自定义多色水平条形图 matplotlib

Question

我想创建一个水平条形图，其中每个条形都有自定义的条纹配色方案（类似于下图）。我熟悉创建普通的 barhplot，但我不知道双色方案如何工作。

尝试过的代码：

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

df = pd.DataFrame({"P1_P2": [0.20],
                   "P1_P3": [0.16],
                   "P2_P5": [0.12],
                   "P3_P5": [0.06],
                   "P3_P6": [0.06],
                   "P5_P6": [0.06]})
df=df.T

fig, ax = plt.subplots(dpi=600, figsize=(4, 4), nrows=1, ncols=1, facecolor=None, edgecolor='black')

df.plot.barh(ax=ax, position=0.50, width=0.3, color=(245/255, 153/255, 145/255, 1.0))
ax.get_legend().remove()

plt.show()

Answer 1

这可以通过在 barh 中使用 facecolor 和 edgecolor 参数来实现。我看到您在 subplots 中使用了以下参数。该图是使用以下解决方案重新创建和生成的：

%matplotlib

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

df = pd.DataFrame({"P1_P2": [0.20],
                   "P1_P3": [0.16],
                   "P2_P5": [0.12],
                   "P3_P5": [0.06],
                   "P3_P6": [0.06],
                   "P5_P6": [0.06]})
df=df.T

fig, ax = plt.subplots(dpi=600, figsize=(4, 4), nrows=1, ncols=1)

df.plot.barh(ax=ax, position=0.50, width=0.3, facecolor="red", 
                     edgecolor="blue")
ax.get_legend().remove()

plt.show()

要在颜色之间添加条纹，可以使用参数 hatch，如下面的代码所述：

%matplotlib

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

df = pd.DataFrame({"P1_P2": [0.20],
                   "P1_P3": [0.16],
                   "P2_P5": [0.12],
                   "P3_P5": [0.06],
                   "P3_P6": [0.06],
                   "P5_P6": [0.06]})
df=df.T

fig, ax = plt.subplots(dpi=600, figsize=(4, 4), nrows=1, ncols=1)

df.plot.barh(ax=ax, position=0.50, width=0.3, facecolor="red", 
                     edgecolor="blue", hatch=r'//')
ax.get_legend().remove()

plt.show()

情节看起来像这样：

如果您想增加或减少线宽，可以使用 plt.rcParams["hatch.linewidth"]

进行更改

plt.rcParams["hatch.linewidth"]= 4

Answer 2

您可以用不同颜色填充每个条形。线条的宽度需要根据特定的情节进行调整。

由于轮廓线使用相同的颜色，因此可以创建仅显示轮廓线的条形副本。

import matplotlib.pyplot as plt
from matplotlib import rcParams
import pandas as pd
from copy import copy

df = pd.DataFrame([0.2 , 0.16, 0.12, 0.06, 0.06, 0.06], index=['P1_P2', 'P1_P3', 'P2_P5', 'P3_P5', 'P3_P6', 'P5_P6'])
first_colors = ['dodgerblue', 'orange', 'green','silver', 'silver', 'dodgerblue']
second_colors = ['crimson', 'crimson', 'dodgerblue', 'dodgerblue', 'green', 'gold']

fig, ax = plt.subplots(dpi=600, figsize=(4, 4), nrows=1, ncols=1, facecolor=None, edgecolor='black')

df.plot.barh(ax=ax, position=0.50, width=0.3, legend=False, linewidth=0)

rcParams['hatch.linewidth'] = 4
for bar, main_color, hatch_color in zip(ax.containers[0], first_colors, second_colors):
    rect = copy(bar)
    rect.set_facecolor('none')
    rect.set_edgecolor('black')
    rect.set_linewidth(2)
    ax.add_patch(rect)
    bar.set_facecolor(main_color)
    bar.set_edgecolor(hatch_color)
    bar.set_hatch('//')
ax.invert_yaxis()
plt.tight_layout()
plt.show()

PS：获取轮廓的更简单版本可能是绘制两次条形的轮廓版本，然后更新首先绘制的条形的颜色和阴影线（第一个条形是存储在 ax.containers[0]).

for _ in range(2):
    df.plot.barh(ax=ax, position=0.50, width=0.3, legend=False, fc='none', ec='black', lw=2)
rcParams['hatch.linewidth'] = 4
for bar, main_color, hatch_color in zip(ax.containers[0], first_colors, second_colors):
    bar.set_linewidth(0)
    bar.set_facecolor(main_color)
    bar.set_edgecolor(hatch_color)
    bar.set_hatch('//')

自定义多色水平条形图 matplotlib

Custom multi-colored horizontal bar chart matplotlib

python

matplotlib

bar-chart

colorbar