python中matplotlib绘图时，画布为正方形，但是里面的图不是正方形，该怎么设置

莫待无花空折枝

# T-SNE可视化
    # 取'Conv1D_1'层 的输出为输出新建为model，采用函数模型
    Conv1D_1_layer_model = Model(inputs=model.input, outputs=model.get_layer('Conv1D_1').output)
    out = Conv1D_1_layer_model.predict(x_train)
    print(out.shape)
    out = out.reshape(7000, -1)
    # np.argmax返回最大数的索引，即可将one-hot编码转换为整数
    color = [np.argmax(i) for i in y_train]
    color = np.stack(color, axis=0)
    n_neighbors = 10  # 一共有多少个类别
    n_components = 2  # 降维成几维 2或者3

    # 创建自定义图像
    fig = plt.figure(figsize=(15,15))  # 指定图像的宽和高
    # t-SNE的最终结果的降维与可视化
    ts = manifold.TSNE(n_components=n_components, init='pca', random_state=0)
    y = ts.fit_transform(out)
    ax1 = fig.add_subplot(3, 1, 2)


    def colormap():
        return mpl.colors.LinearSegmentedColormap.from_list \
            ('cmap', ['#8B0000', '#FF6A6A', '#00FFFF', '#F4A460', '#00CDCD', '#0000FF', '#40E0D0', '#98F5FF', '#FFFF00',
                      '#FF0000'], 256)
        # ('cmap', ['#FFFFFF', '#98F5FF', '#00FF00', '#FFFF00','#FF0000', '#8B0000'], 256) #white, CadetBlue1, Green1, Yellow1, Red,DarkRed


    # cm = plt.cm.Spectral
    cm = colormap()
    plt.scatter(y[:, 0], y[:, 1], c=color, cmap=cm)

    ax1.set_title('conv1', fontsize=15)
    plt.show()