在 Matplotlib 中新增子圖
本教程介紹瞭如何使用 Python 中的 matplotlib.pyplot.figure.add_subplot() 方法為 Matplotlib 圖新增子圖。
使用 matplotlib.pyplot.figure.add_subplot() 方法在圖中新增子圖
import matplotlib.pyplot as plt
fig=plt.figure(figsize=(5,5))
x=[1,2,3,4,5]
y=[2,4,5,1,2]
subplot=fig.add_subplot(1,1,1)
subplot.plot(x, y)
fig.suptitle("Figure with only one Subplot")
plt.show()
輸出：
它生成的圖中只有一個子圖。該圖可以解釋為一個包含所有子圖和圖中其他元素的高階元素。我們首先初始化大小為 (5,5) 的圖物件，然後使用 add_subplot() 方法向圖中新增一個子圖。add_subplot() 方法的第 1 個引數代表圖中子圖的行數，第 2 個引數代表子圖的列數，第 3 個引數代表子圖在網格中的位置。
這裡，fig.add_subplot(1,1,1) 將在圖中 1 X 1 網格的第一個位置新增一個子圖。另外，我們也可以使用 fig.add_subplot(111) 來實現同樣的結果。add_subplot() 方法返回一個 axes 物件，它可以用來繪製和新增子圖的細節。
import matplotlib.pyplot as plt
fig=plt.figure(figsize=(8,6))
ax_1=fig.add_subplot(121)
ax_1.text(0.3, 0.5, 'subplot(121)')
ax_2=fig.add_subplot(222)
ax_2.text(0.3, 0.5, 'subplot(222)')
ax_3=fig.add_subplot(224)
ax_3.text(0.3, 0.5, 'subplot(224)')
fig.suptitle("Figure with multiple Subplots")
plt.show()
輸出：
這裡，我們有一個包含三個子圖的圖。第一個子圖是網格中的第 1 個數字，有 1 行 2 列。第 2 個子圖代表網格中的第 2 個數字，有兩行兩列。同樣，第三行代表網格中第四個位置的子圖，有兩行兩列。
使用 Matplotlib.pyplot.subplots() 建立帶有子圖的圖
Matplotlib.pyplot.subplots() 也可以給圖新增子圖。我們將行數和列數作為引數傳遞給該方法，該方法返回一個圖物件和軸物件，可以用來操作圖形。
import matplotlib.pyplot as plt
fig,ax=plt.subplots(2,1)
ax[0].text(0.3,0.5,"1st Subplot")
ax[0].set_xticks([])
ax[0].set_yticks([])
ax[1].text(0.3,0.5,"2nd Subplot")
ax[1].set_xticks([])
ax[1].set_yticks([])
fig.suptitle('Figure with 2 subplots',fontsize=16)
plt.show()
輸出：
它使用 subplots() 方法在兩行一列的網格中生成一個有兩個子圖的圖形。第 1 個子圖或頂部的軸可以用 ax 列表中的第 1 個元素進行操作，第 2 個子圖或底部的軸可以用 ax 列表中的第 2 個元素進行操作。
import matplotlib.pyplot as plt
x=[1,2,3,4,5]
y1=[i**2 for i in x ]
y2=[4*i+1 for i in x]
fig,ax=plt.subplots(2,1)
ax[0].plot(x,y1)
ax[0].set_xlabel("X")
ax[0].set_ylabel("Y")
ax[1].plot(x,y2)
ax[1].set_xlabel("X")
ax[1].set_ylabel("Y")
plt.tight_layout()
plt.show()
輸出：
它生成一個有兩個子圖的圖形，在上面的子圖中繪製 x^2 函式，在下面的子圖中繪製 4x+1 函式。
Suraj Joshi is a backend software engineer at Matrice.ai.
LinkedIn