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

莫待无花空折枝

1. # T-SNE可视化
   
2.     # 取'Conv1D_1'层 的输出为输出新建为model，采用函数模型
   
3.     Conv1D_1_layer_model = Model(inputs=model.input, outputs=model.get_layer('Conv1D_1').output)
   
4.     out = Conv1D_1_layer_model.predict(x_train)
   
5.     print(out.shape)
   
6.     out = out.reshape(7000, -1)
   
7.     # np.argmax返回最大数的索引，即可将one-hot编码转换为整数
   
8.     color = [np.argmax(i) for i in y_train]
   
9.     color = np.stack(color, axis=0)
   
10.     n_neighbors = 10  # 一共有多少个类别
    
11.     n_components = 2  # 降维成几维 2或者3
    
12. 
    
13.     # 创建自定义图像
    
14.     fig = plt.figure(figsize=(15,15))  # 指定图像的宽和高
    
15.     # t-SNE的最终结果的降维与可视化
    
16.     ts = manifold.TSNE(n_components=n_components, init='pca', random_state=0)
    
17.     y = ts.fit_transform(out)
    
18.     ax1 = fig.add_subplot(3, 1, 2)
    
19. 
    
20. 
    
21.     def colormap():
    
22.         return mpl.colors.LinearSegmentedColormap.from_list \
    
23.             ('cmap', ['#8B0000', '#FF6A6A', '#00FFFF', '#F4A460', '#00CDCD', '#0000FF', '#40E0D0', '#98F5FF', '#FFFF00',
    
24.                       '#FF0000'], 256)
    
25.         # ('cmap', ['#FFFFFF', '#98F5FF', '#00FF00', '#FFFF00','#FF0000', '#8B0000'], 256) #white, CadetBlue1, Green1, Yellow1, Red,DarkRed
    
26. 
    
27. 
    
28.     # cm = plt.cm.Spectral
    
29.     cm = colormap()
    
30.     plt.scatter(y[:, 0], y[:, 1], c=color, cmap=cm)
    
31. 
    
32.     ax1.set_title('conv1', fontsize=15)
    
33.     plt.show()

复制代码

rsj0315

那就适当修改两个坐标的范围调成看着等长吧