1 import matplotlib.pyplot as plt
2
3 # example 1:
4 ###############################
5 plt.figure(figsize=(6, 4))
6 # plt.subplot(n_rows, n_cols, plot_num)
7 plt.subplot(2, 2, 1)
8 plt.plot([0, 1], [0, 1])
9
10 plt.subplot(2, 2, 2)
11 plt.plot([0, 1], [0, 2])
12
13 plt.subplot(2, 2, 3)
14 plt.plot([0, 1], [0, 3])
15
16 plt.subplot(2, 2, 4)
17 plt.plot([0, 1], [0, 4])
18
19 plt.tight_layout()
20
21 # example 2:
22 ###############################
23 plt.figure(figsize=(6, 4))
24 # figure splits into 2 rows, 1 col, plot to the 1st sub-fig
25 plt.subplot(2, 1, 1)
26 plt.plot([0, 1], [0, 1])
27 # figure splits into 2 rows, 3 col, plot to the 4th sub-fig
28 plt.subplot(2, 3, 4)
29 plt.plot([0, 1], [0, 2])
30 # figure splits into 2 rows, 3 col, plot to the 5th sub-fig
31 plt.subplot(2, 3, 5)
32 plt.plot([0, 1], [0, 3])
33 # figure splits into 2 rows, 3 col, plot to the 6th sub-fig
34 plt.subplot(2, 3, 6)
35 plt.plot([0, 1], [0, 4])
36
37 plt.tight_layout()
38
39 plt.show()
1 import matplotlib.pyplot as plt
2 import matplotlib.gridspec as gridspec
3
4 # method 1: subplot2grid
5 ##########################
6 plt.figure()
7 #整个figure为 3*3
8 ax1 = plt.subplot2grid((3, 3), (0, 0), colspan=3) #ax1起始是第一行第一列，占一行三列
9 #plt.subplot2grid(shape, loc, rowspan=1, colspan=1, fig=None, **kwargs)
10 #参数解释：
11 # shape : sequence of 2 ints
12 # loc : sequence of 2 ints
13 # rowspan : int Number of rows for the axis to span to the right.
14 # colspan : int Number of columns for the axis to span downwards.
15 # rowspan、colspan默认值为1
16 ax1.plot([1, 2], [1, 2])
17 ax1.set_title('ax1_title')
18 ax2 = plt.subplot2grid((3, 3), (1, 0), colspan=2) #ax2起始为第二行第一列，占一行两列
19 ax3 = plt.subplot2grid((3, 3), (1, 2), rowspan=2) #ax3起始为第二行第三列，占两行一列
20 ax4 = plt.subplot2grid((3, 3), (2, 0)) #ax4起始为第三行第一列，占一行一列，散点图
21 ax4.scatter([1, 2], [2, 2])
22 ax4.set_xlabel('ax4_x')
23 ax4.set_ylabel('ax4_y')
24 ax5 = plt.subplot2grid((3, 3), (2, 1)) #ax5起始为第三行第二列，占一行一列
25
26 # method 2: gridspec
27 #########################
28 plt.figure()
29 gs = gridspec.GridSpec(3, 3)
30 # use index from 0
31 ax6 = plt.subplot(gs[0, :])
32 ax7 = plt.subplot(gs[1, :2])
33 ax8 = plt.subplot(gs[1:, 2])
34 ax9 = plt.subplot(gs[-1, 0])
35 ax10 = plt.subplot(gs[-1, -2])
36
37 #这里理解矩阵切片相关知识很好理解，矩阵切片相关知识，参考博客 https://blog.csdn.net/u014453898/article/details/73430907
38
39 # method 3: easy to define structure
40 ####################################
41 f, ((ax11, ax12), (ax13, ax14)) = plt.subplots(2, 2, sharex = True, sharey = True) #不同subplot共享x，y轴
42 ax11.scatter([1,2], [1,2])
43
44 plt.tight_layout()
45 plt.show()
