一、概述

plt.subplot是matplotlib庫(kù)中的一個(gè)函數(shù)，它的作用是在單個(gè)圖像中創(chuàng)建多個(gè)子圖。通過(guò)自定義子圖位置，我們可以將多組數(shù)據(jù)在同一個(gè)圖像中展示出來(lái)，從而更直觀地比較各組數(shù)據(jù)的差異。這一功能在數(shù)據(jù)分析與數(shù)據(jù)可視化中有著廣泛的應(yīng)用。

二、使用方法

在使用plt.subplot之前，需要先導(dǎo)入matplotlib庫(kù)。plt.subplot函數(shù)共有三個(gè)參數(shù)：子圖行數(shù)、子圖列數(shù)和子圖位置。其中，子圖位置是一個(gè)整數(shù)，其取值范圍為1至子圖行數(shù)與子圖列數(shù)的積。

import matplotlib.pyplot as plt
plt.figure(figsize=(8,6)) # 畫(huà)布大小
plt.subplots_adjust(hspace=0.4) # 子圖間距
plt.subplot(2,2,1) # 創(chuàng)建第1個(gè)子圖
plt.plot([0,1], [0,1])
plt.title('subplot 1')
plt.subplot(2,2,2) # 創(chuàng)建第2個(gè)子圖
plt.plot([1,0], [0,1])
plt.title('subplot 2')
plt.subplot(2,1,2) # 創(chuàng)建第3個(gè)子圖
plt.plot([0,1], [0,1])
plt.title('subplot 3')
plt.show()

運(yùn)行以上代碼，即可創(chuàng)建一個(gè)包含3個(gè)子圖的圖像。其中第1和第2個(gè)子圖都位于第1行，第3個(gè)子圖位于第2行。

三、子圖位置的設(shè)定

除了直接指定整數(shù)作為子圖位置，我們還可以使用子圖位置的另一種設(shè)定方式——使用plt.subplots中的gridspec參數(shù)。gridspec參數(shù)是一個(gè)2元素元組，分別指定子圖行數(shù)和列數(shù)。通過(guò)調(diào)用gridspec子命令來(lái)設(shè)置子圖位置。代碼如下：

import matplotlib.gridspec as gridspec
import matplotlib.pyplot as plt
plt.figure(figsize=(8,6))
gs = gridspec.GridSpec(2,2) # 兩行兩列
ax1 = plt.subplot(gs[0,0]) # 第1個(gè)子圖
ax1.plot([0,1], [0,1])
ax1.set_title('subplot 1')

ax2 = plt.subplot(gs[0,1]) # 第2個(gè)子圖
ax2.plot([1,0], [0,1])
ax2.set_title('subplot 2')

ax3 = plt.subplot(gs[1,:]) # 第3個(gè)子圖
ax3.plot([0,1], [0,1])
ax3.set_title('subplot 3')
plt.show()

運(yùn)行以上代碼，同樣可以創(chuàng)建一個(gè)有3個(gè)子圖的圖像。不同之處在于，這次我們使用了一種更靈活的設(shè)定方法，即通過(guò)gridspec.GridSpec創(chuàng)建網(wǎng)格，通過(guò)plt.subplot指定子圖位置。

四、多行多列圖形

除了使用plt.subplot來(lái)創(chuàng)建子圖，我們還可以使用plt.subplots來(lái)一次性創(chuàng)建多行多列的圖形。代碼如下：

import matplotlib.pyplot as plt
fig,axes = plt.subplots(nrows=2, ncols=2, figsize=[8,6])
axes[0,0].plot([0,1], [0,1])
axes[0,0].set_title('subplot 1')
axes[0,1].plot([1,0], [0,1])
axes[0,1].set_title('subplot 2')
axes[1,0].plot([0,1], [1,0])
axes[1,0].set_title('subplot 3')
axes[1,1].plot([0,1], [0,0])
axes[1,1].set_title('subplot 4')
plt.show()

這段代碼創(chuàng)建了一個(gè)2×2的圖形，每個(gè)子圖都有一個(gè)對(duì)應(yīng)的axes變量名用于進(jìn)行進(jìn)一步的設(shè)定。

五、子圖刻度的設(shè)定

在使用plt.subplot或plt.subplots創(chuàng)建子圖后，我們還希望設(shè)定每個(gè)子圖的橫、縱坐標(biāo)軸上的刻度。這可以通過(guò)子圖對(duì)象的axis()方法或set_xticks()、set_yticks()方法實(shí)現(xiàn)。代碼如下：

import matplotlib.pyplot as plt
plt.figure(figsize=(8,6))
plt.subplots_adjust(hspace=0.4)

plt.subplot(2,2,1)
plt.plot([0,1], [0,1])
plt.title('subplot 1')
plt.xticks([0,0.5,1]) # 水平刻度線
plt.yticks([0,0.5,1]) # 垂直刻度線

plt.subplot(2,2,2)
plt.plot([1,0], [0,1])
plt.title('subplot 2')
plt.xticks([0,0.5,1]) # 水平刻度線
plt.yticks([0,0.5,1]) # 垂直刻度線

plt.subplot(2,1,2)
plt.plot([0,1], [0,1])
plt.title('subplot 3')
plt.xticks([0,0.5,1]) # 水平刻度線
plt.yticks([0,0.5,1]) # 垂直刻度線
plt.show()

運(yùn)行以上代碼，即可看到每個(gè)子圖都有自己的刻度線。

六、子圖標(biāo)題、圖形標(biāo)題和軸標(biāo)簽

除了刻度線外，我們還可以通過(guò)子圖對(duì)象的set_xlabel()、set_ylabel()和set_title()方法來(lái)設(shè)定自己的軸標(biāo)簽和標(biāo)題。代碼如下：

import matplotlib.pyplot as plt
plt.figure(figsize=(8,6))
plt.subplots_adjust(hspace=0.7)

plt.subplot(2,2,1)
plt.plot([0,1], [0,1])
plt.title('subplot 1')
plt.xlabel('x1')
plt.ylabel('y1')

plt.subplot(2,2,2)
plt.plot([1,0], [0,1])
plt.title('subplot 2')
plt.xlabel('x2')
plt.ylabel('y2')

plt.subplot(2,1,2)
plt.plot([0,1], [0,1])
plt.title('subplot 3')
plt.xlabel('x3')
plt.ylabel('y3')
plt.show()

運(yùn)行以上代碼，即可看到每個(gè)子圖都有自己的軸標(biāo)簽和標(biāo)題。

七、小結(jié)

本文詳細(xì)解釋了plt.subplot函數(shù)的使用方法和設(shè)定技巧，并通過(guò)代碼示例進(jìn)行了演示。希望本文能夠?qū)ψx者有所幫助，激發(fā)大家在數(shù)據(jù)分析和可視化中更加靈活地利用matplotlib庫(kù)。