matplotlib imshow 显示比例修复为正方形

Question

我的代码可以根据以下示例生成绘图。如果可能，我想强制输出图为128x128 正方形，而不是64x128。我的数据旨在被视为一个正方形，即使它的矩阵没有表现出 NxN 属性。

谢谢。

import numpy as np
import pylab as pl

my_matrix = []
for x in range(128):
    row = []
    for value in range(64):
        row.append( float(value) / 63 )
    my_matrix.append(row)

array = np.matrix(my_matrix)
pl.axes()
pl.imshow(array, interpolation='none', cmap='jet', origin='lower')
pl.colorbar(shrink=0.95)
pl.xticks(())
pl.yticks(())
pl.show()

Answer 1

您可以按如下方式定义轴纵横比：

pl.axes().set_aspect(0.5)

因子 0.5 补偿了数据的纵横比 128:64 = 2:1。

有关更多信息，您可以查看this answer。