matplotlib使用GridSpec調整子圖位置大小 (非對稱的子圖)

用matplotlib.pyplot的subplots命令能夠很方便的畫對稱的子圖,可是若是要畫非對稱的子圖(以下)就須要用GridSpec命令來控制子圖的位置和大小:
html

而上圖的結構能夠用一下兩種方式畫:python

import matplotlib.pyplot as plt
from matplotlib.gridspec import GridSpec

fig = plt.figure(1)
gs = GridSpec(3, 3)

ax1 = plt.subplot(gs[0, :])
ax2 = plt.subplot(gs[1, :2])
ax3 = plt.subplot(gs[1:, 2])
ax4 = plt.subplot(gs[2, 0])
ax5 = plt.subplot(gs[2, 1])

或者函數

ax1 = plt.subplot2grid((3, 3), (0, 0), colspan=3)
ax2 = plt.subplot2grid((3, 3), (1, 0), colspan=2)
ax3 = plt.subplot2grid((3 ,3), (1, 2), rowspan=2)
ax4 = plt.subplot2grid((3, 3), (2, 0))
ax5 = plt.subplot2grid((3, 3), (2, 1))

更多間隔設置能夠參考gridspec的官方文檔。spa

PS:若是是讀取圖片做爲子圖的話建議用Pillow包的Image函數讀取,而非自帶的imread函數,對圖像的調整會方便不少:code

ax1 = fig.add_subplot(gs[:, 0])
img = Image.open("your image")
ax1.imshow(img)
相關文章
相關標籤/搜索