问如何在生成的热图图的某个地方插入一个小正方形标记

EN

    from matplotlib.patches import Rectangle
    import numpy as np
    import matplotlib.pyplot as plt
    from sklearn.metrics import r2_score, median_absolute_error
    import os
    import matplotlib.cm as cm
    from mpl_toolkits import axes_grid1
    import matplotlib.pyplot as plt
    #import seaborn as sns
    import matplotlib.pyplot as plt
    import matplotlib.colors as mcolors
    import matplotlib.colors
    import matplotlib.colors as colors
    
    
    data = np.random.rand(10, 150, 150)
    data = data.reshape(-1, 1)
    
    
    property = "Sale"
    
       
    pmin = data.min()
    pmax = data.max()
    
    v = np.linspace(round(pmin,3), round(pmax,3),15, endpoint=True)
    v = [round(x,3) for x in v] 
    
    fig, ax = plt.subplots(2, 5, figsize=(160, 80))
    row_count = 0
    col_count = 0
    for i in range(10):
    
        sub_plot_data = data[(i)*(150*150):(i+1)*150*150]
        
       
        x = 150
        y = 150
        #--------------------------- Define the map boundary ---------------------- 
        xmin = 1258096.6
        xmax = 1291155.0
        ymin = 11251941.6
        ymax = 11285000.0
        
        pmin = min(sub_plot_data)
        pmax = max(sub_plot_data) 
    
        
        # ---------------------------  define color bar for Discrete  color 
        bounds = np.linspace(-1, 1, 10)
        Discrete_colors = plt.get_cmap('jet')(np.linspace(0,1,len(bounds)+1))
        # create colormap without the outmost colors
        cmap = mcolors.ListedColormap(Discrete_colors[1:-1]) # 


        actual_2d = np.reshape(sub_plot_data,(y,x)) 
        
        im1 = ax[row_count, col_count].imshow(actual_2d, interpolation=None, cmap=cmap, 
        extent=(xmin, xmax, ymin, ymax), vmin=pmin, vmax=pmax)      
        plt.text(actual_2d[62, 62], actual_2d[62, 62], '%s' % 'Sale_1', 
        horizontalalignment='center', verticalalignment='center', color= 'black', fontsize= 90)
    
        
        ax[row_count, col_count].set_title("Sale_Stores-%s - L: %s"%(i+1, layer), 
        fontsize=130, pad=44, x=0.5, y=0.999) # new
    
        ax[row_count, col_count].set_aspect('auto')
        ax[row_count, col_count].tick_params(left=False, labelleft=False, top=False, 
        labeltop=False, right=False, labelright=False, bottom=False, labelbottom=False) # new
        #ax[row_count, col_count] = plt.gca()
        plt.gca().add_patch(Rectangle((60, 60), 3, 3, edgecolor='black', 
        facecolor='black',fill=True,lw=2))
        ax[row_count, col_count].add_patch(plt.text(62, 62, '%s' % 'Sale_1', 
       horizontalalignment='center', verticalalignment='center', color= 'black', fontsize= 90))
    
        col_count +=1
        
        if col_count == 5:
            row_count +=1  
            col_count =0
    
           
       
    fig.tight_layout(h_pad=10) 
    plt.subplots_adjust(left=0.02,
                    bottom=0.1, 
                    right=0.91, 
                    top=0.8, 
                    wspace=0.1, 
                    hspace=0.2)
    
      
    cbaxes = fig.add_axes([0.94, 0.05, 0.02, 0.8]) 
    cbar = fig.colorbar(im1, ax=ax.ravel().tolist(), ticks=v, extend='both', cax =cbaxes)
    cbar.ax.tick_params(labelsize=70) 
    #cbar.set_ticks(v)
    cbar.ax.set_yticklabels([i for i in v], fontsize=120)
    
    
    output_dir = r"D/test"
    plot_dir = os.path.join(output_dir, reservoir_property)
    if not os.path.exists(plot_dir):
        os.makedirs(plot_dir)
    fig.savefig(r"%s/per_allmodel.png"%(plot_dir))