Получить ось из стволового участка

Question

Я хотел бы изменить ось графика ствола, например, изменить цвет сетки с помощью ax.grid(color='gray', axis='y') на графике ствола пример . Как бы я это сделал?

import matplotlib.pyplot as plt
import numpy as np

# returns 10 evenly spaced samples from 0.1 to 2*PI
x = np.linspace(0.1, 2 * np.pi, 10)

markerline, stemlines, baseline = plt.stem(x, np.cos(x), '-.')

# setting property of baseline with color red and linewidth 2
plt.setp(baseline, color='r', linewidth=2)

Answer 1

Кроме того, вы можете просто использовать plt без определения ax. Хотя я предпочитаю последнее (ваш ответ).

plt.grid(color='gray', axis='y')

Answer 2

За Как получить экземпляр Axp matplotlib для построения на? , другой вариант добавляет это в конце:

ax = plt.gca()
ax.grid(color='gray', axis='y')

(ht ImportanceOfBeingErnest)

Answer 3

Сначала определите ось, используя plt.subplots(), затем назовите ax.stem вместо plt.stem:

fig, ax = plt.subplots()
markerline, stemlines, baseline = ax.stem(x, np.cos(x), '-.')
ax.grid(color='gray', axis='y')