Как добавить вертикальные линии сетки на диаграмму matplotlib?

Question

У меня есть следующий кусок кода

import matplotlib.pyplot as plt
import numpy as np

[...]

    def display(self):
        t = self.stateLabels

        fig, axs = plt.subplots(4, 1, sharex=True)
        fig.subplots_adjust(hspace=0)

        axs[0].plot(t, self.production)
        axs[0].title.set_text('Production')
        axs[0].frameon = True

        axs[1].plot(t, self.consumption)
        axs[1].title.set_text('Consumption')

        axs[2].plot(t, self.investment)
        axs[2].title.set_text('Investment')

        axs[3].plot(t, self.gdp)
        axs[3].title.set_text('GDP')

        plt.show()

Это создает следующую диаграмму:

Я хочу добавить вертикальные линии к осям, чтобы диаграмма выглядела следующим образом:

Возможно ли это сделать? Если да, то как?

Answer 1

Просмотр документации matplotlib на ax.axvline Это должно дать вам возможность рисовать вертикальные линии в указанной позиции x и с указанными начальной и конечной точкой.