使用Python绘制轮廓图的纵横比错误（Matplotlib）

Question

我导入了一个拟合文件，基本上想将其绘制为等高线图。问题出现在宽高比上。图像尺寸为320x240，因为我不希望图片被拉伸，所以我希望图像的比例也是320/240。

然而，Python将它强制化为方形，这就是我使用plt.axis('equal')的原因 然而，这会在绘图区域产生白色区域，我不知道如何摆脱它们......

有什么想法吗？

非常感谢!!

import numpy
import math
from astropy.io import fits
import matplotlib.pyplot as plt

scidatafile=fits.open('24-02-2015-PSF-OD00-iris15-30pin.fits')
scidata=scidatafile[0].data
oneframe=scidata[0]
oneframe[oneframe<0]=0
yvec=numpy.linspace(0,240,240)
xvec=numpy.linspace(0,320,320)
X,Y=numpy.meshgrid(xvec,yvec)

它出现了

plt.contourf(X,Y,oneframe)
plt.axis('equal')
plt.savefig('this.pdf')
plt.show()

Answer 1

使用pylab，你可以这样做：

import pylab as pl
import numpy as np

yvec=np.linspace(0,240,240)
xvec=np.linspace(0,320,320)
X,Y=np.meshgrid(xvec,yvec)

Z = X**2 + Y**2 # function to plot

sx = 8
sy = sx*float(yvec.size)/float(xvec.size) # reset y-scale to match array dimensions

fig, ax = pl.subplots(figsize=(sx, sy))
ax.contourf(X,Y,Z)

在简单的matplotlib中，我认为使用gcf()来操纵图形可以获得相同的效果。

Answer 2

首先，您应该发布一个最小的工作示例。当您使用未提供的数据时，我只使用了一些自己的虚拟数据。

基本上问题是在任何数据在轴之前设置纵横比。我不知道为什么，如果是我，我会把它标记为bug而不是功能。因此，在调用contourf后设置宽高比。

我也不明白为什么aspect = 1不适用于contourf，而它适用于imshow。

    from numpy import *
    from matplotlib.pyplot import *

    # dummy data preparations

    x = linspace( -3.0, 3.0, 100 )
    y = linspace( -2.0, 2.0, 100 )

    def V( x, y=None ):
        if y == None:
            return 1.0/x
        xv,yv = meshgrid( x, y, sparse=False, indexing='xy' )
        return 1.0 / sqrt(xv**2+yv**2)

    rlev = sqrt(18.) / 2.**linspace( 0,4,10 )
    pic = V(x,y)

    # your try to plot it which results in white borders
    fig = figure()
    title('old version')
    contourf( x, y, pic, levels=V(rlev) )
    axis('equal')
    tight_layout()
    savefig('old.png')

需要10个回复才能发布图片：'old.png'

    # method without white borders
    fig = figure()
    title('new version')
    ax = subplot(1,1,1)
    contourf( x, y, pic, levels=V(rlev) )
    ax.set_aspect('equal')
    tight_layout()
    savefig('new.png')
    show()

需要10个回复才能发布图片：'new.png'