從主干圖獲取軸

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

首先使用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')

Answer 3

每如何獲得一個matplotlib Axes實例進行繪制？ ，另一個選擇是在末尾添加：

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

（ht ImportanceOfBeingErnest）