Блог

Я хочу создать горизонтальную линейчатую диаграмму, где каждая полоса имеет пользовательскую цветовую схему в полоску (аналогично рисунку ниже). Я знаком с созданием обычного штрихплота, но я не знаю, как будет работать двухцветная схема.

Попробовал код:

 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()
 

Вы можете штриховать каждый из стержней разными цветами. Ширину линий необходимо адаптировать в зависимости от конкретного участка.

Поскольку для штриховки используется тот же цвет, что и для контуров, можно создать копию полосы, которая отображает только контур.

 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('//')
 

Это может быть достигнуто с помощью аргументов facecolor и edgecolor внутри barh . Я вижу, что вы использовали следующие аргументы в 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