Добавить вертикальную линию к мульти горизонтальной гистограмме

Question

Я создаю мульти-гистограмму в виде горизонтальной гистограммы, и сейчас мне нужно аннотировать (в основном, создавать вертикальную линию) каждую горизонтальную гистограмму в определенном месте (на основе значения оси X)для каждой оси y, где ось y является категориальной (имена), а ось x - числовой (целое число) s.

Я посмотрел в axis.vlines, но не смог заставить его работать.

import seaborn as sns
import matplotlib.pyplot as plt
crashes = sns.load_dataset("car_crashes").sort_values("total", ascending=False)
crashes['max_range'] = crashes['total'] * 0.85
sns.set_color_codes("muted")
sns.set(style="whitegrid")
sns.barplot(x="total", y="abbrev", data=crashes, label="", color="r")
sns.barplot(x="max_range", y="abbrev", data=crashes, label="", color="y")
sns.barplot(x="alcohol", y="abbrev", data=crashes,label="normal range", color="g")

#dummy data for the "vertical lines" i want to plot
crashes['actual'] = crashes['alcohol'] * 1.85

Приведенный выше код создает график, подобный следующему:

https://seaborn.pydata.org/examples/horizontal_barplot.html

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

Answer 1

axis.vlines достаточно для работы. Я делаю это, сначала извлекая точки y для меток гистограммы. Затем я делаю словарь значений х для этих точек. чем я использую axis.vlines, чтобы нарисовать красную линию на столбцах.

import seaborn as sns
import matplotlib.pyplot as plt

crashes = sns.load_dataset("car_crashes").sort_values("total", ascending=False)
crashes['max_range'] = crashes['total'] * 0.85
sns.set_color_codes("muted")
sns.set(style="whitegrid")
# Store the returned axes in a variable
ax = sns.barplot(x="total", y="abbrev", data=crashes, label="", color="r")
ax = sns.barplot(x="max_range", y="abbrev", data=crashes, label="", color="y")
ax = sns.barplot(x="alcohol", y="abbrev", data=crashes,label="normal range", color="g")

#dummy data for the "vertical lines" i want to plot
crashes['actual'] = crashes['alcohol'] * 1.85


#### MY ADDITIONS ####

# Form dictionary of bar chart keys (i.e. Y axis data, here it is "abbrev") to
# corresponding y and x points
y_labs = list(ax.get_yticklabels())
y_tic_pos = list(ax.get_yticks())
y_tick_vals = {}
for i in range(len(y_tic_pos)):
    y_tick_vals[y_labs[i].get_text()] = y_tic_pos[i]
x_points = {lab:crashes[crashes["abbrev"] == lab]["actual"].values[0] for lab in y_tick_vals}

# for each of the relevant y axis, draw a vertical line
for key in y_tick_vals:
    c_y = y_tick_vals[key]
    c_x = x_points[key]
    # I just did some trial and error to find out that each bar is 0.5 wide;
    # this may not be the case for other plots.
    c_ymin = c_y - 0.25
    c_ymax = c_y + 0.25

    ax.vlines(c_x, c_ymin, c_ymax, colors="r")

plt.show()

Должен быть более элегантный способ сделать это; это первый прототип, который я смог придумать.

---

P.S. Вы говорите, что пытались использовать axis.vlines, но не включили код, чтобы показать нам вашу попытку. Я думаю, что это было одной из причин, по которой ваш вопрос был опущен.