pythonfiguresubplot

Center the third subplot in the middle of second row python


I have a figure consisting of 3 subplots. I would like to locate the last subplot in the middle of the second row. Currently it is located in the left bottom of the figure. How do I do this? I cannot find the answer on stack overflow.

    fig, axes = plt.subplots(nrows=2, ncols=2, figsize=(13,10))
    ax= axes.flatten()
    ax[0].plot(vDT, np.cumsum(mWopt0[asset0,:])*percentage/iTT, label= 'COAL, c = 0')
    ax[0].legend()
    ax[0].set_title('Proportion in most invested stock')
    ax[1].plot(vDT, np.cumsum(mWopt01[asset01,:])*percentage/iTT, label= 'OINL, c = 0.1')
    ax[1].plot(vDT, np.cumsum(mWopt03[asset03,:])*percentage/iTT, label= 'OINL, c = 0.3')
    ax[1].plot(vDT, np.cumsum(mWopt05[asset05,:])*percentage/iTT, label= 'OINL, c = 0.5')
    ax[1].plot(vDT, np.cumsum(mWopt2[asset2,:])*percentage/iTT, label= 'OINL, c = 2')
    ax[1].plot(vDT, np.cumsum(mWopt5[asset5,:])*percentage/iTT, label= 'OINL, c = 5')
    ax[1].plot(vDT, np.cumsum(mWopt10[asset10,:])*percentage/iTT, label= 'OINL, c = 10')
    ax[1].legend()
    ax[1].set_title('Proportion in most invested stock')
    ax[2].plot(vDT, np.cumsum(mWopt01[index,:])*percentage/iTT, label= 'c = 0')
    ax[2].plot(vDT, np.cumsum(mWopt01[index,:])*percentage/iTT, label= 'c = 0.1')
    ax[2].plot(vDT, np.cumsum(mWopt03[ index,:])*percentage/iTT, label= 'c = 0.3')
    ax[2].plot(vDT, np.cumsum(mWopt05[index,:])*percentage/iTT, label= 'c = 0.5')
    ax[2].plot(vDT, np.cumsum(mWopt2[index,:])*percentage/iTT, label= 'c = 2')
    ax[2].plot(vDT, np.cumsum(mWopt5[index,:])*percentage/iTT, label= 'c = 5')
    ax[2].plot(vDT, np.cumsum(mWopt10[index,:])*percentage/iTT, label= 'c = 10')
    ax[2].legend()
    ax[2].set_title('Proportion invested in index')
    ax[0].set_ylabel('Expanding window weight')
    ax[1].set_ylabel('Expanding window weight')
    ax[2].set_ylabel('Expanding window weight')
    ax[3].remove()
    fig.autofmt_xdate(bottom=0.2, rotation=75, ha='right')
    plt.savefig('NSE_por_unrestricted_mostweightSI.jpg', bbox_inches='tight')
    plt.show()

Solution

  • matplotlib.gridspec.Gridspec solves your problem, and can be passed to plt.subplot. In this answer, you can see that a 4x4 grid can be used to position a plot in the middle easily:

    import matplotlib.gridspec as gridspec
    import matplotlib.pyplot as plt
    
    gs = gridspec.GridSpec(4, 4)
    
    ax1 = plt.subplot(gs[:2, :2])
    ax1.plot(range(0,10), range(0,10))
    
    ax2 = plt.subplot(gs[:2, 2:])
    ax2.plot(range(0,10), range(0,10))
    
    ax3 = plt.subplot(gs[2:4, 1:3])
    ax3.plot(range(0,10), range(0,10))
    
    plt.show()
    

    You can check out the demos for gridspec here: https://matplotlib.org/tutorials/intermediate/gridspec.html#sphx-glr-tutorials-intermediate-gridspec-py

    The only problem is that you are using the fig, axes = pattern, which I don't see being typically used with Gridspec. You would need to refactor a bit.