使用 matplotlib.pyplot、imshow() 和 savefig() 以全分辨率绘图?

Plotting at full resolution with matplotlib.pyplot, imshow() and savefig()?

我有一个中型阵列(例如 1500x3000),我想按比例绘制它,因为它是一个图像。但是,垂直和水平尺度非常不同。为简化起见,假设有一个 meter/row 和 10/列。然后该图应产生一个图像,即 c。 1500x30000。我将 kwarg extent 用于尺度和 aspect = 1 以避免变形。 通过使用绘图 windows (QT4) 和 imshow() 或通过使用 savefig(),我从未成功地以全分辨率生成图像。

我已经查看了 here, , or here and there or there 中指出的许多提议的解决方案,以防它是一个错误。我已经更改了我的 matplotlibrc 并将其放在 ~/.config/matplotlib 中以尝试强制我的 display / savefig 选项但无济于事。我也尝试过 pcolormesh() 但没有成功。我使用 python 2.7 和 matplotlib 1.3 来自 Ubuntu 14.04 和 QT4Agg 作为后端。我也尝试过 TkAgg,但它很慢并且给出了相同的结果。我的印象是在 x 轴上分辨率是正确的,但它在垂直方向上肯定是下采样的。这是一段模拟我的问题的代码。

import numpy as np
import matplotlib.pyplot as plt
import matplotlib.colors

R, C = 1500, 3000
DATA = np.random.random((R, C))
DATA[::2, :] *= -1  # make every other line negative
Yi, Xi = 1, 10 # increment
CMP = 'seismic'
ImageFormat ='pdf'
Name = 'Image'


DataRange = (np.absolute(DATA)).max() # I want my data centred on 0
EXTENT = [0, Xi*C, 0 ,Yi*R]
NORM = matplotlib.colors.Normalize(vmin =-DataRange, vmax= DataRange, clip =True)

for i in range(1,4):
    Fig=plt.figure(figsize=(45, 10), dpi = 100*i, tight_layout=True)
    Fig.suptitle(Name+str(i)+'00DPI')
    ax = Fig.add_subplot(1, 1, 1)
    Plot = ax.imshow(DATA, cmap=plt.get_cmap(CMP), norm = NORM, extent = EXTENT, aspect = 1, interpolation='none') 
    ax.set_xlabel('metres')
    ax.set_ylabel('metres')
    Fig.savefig(Name+str(i)+'00DPI.'+ImageFormat,  format = ImageFormat, dpi = Fig.dpi)
plt.close()

在 imshow() 中,插值 = 'none' 或 'nearest' 或 'bilinear' 由于某种原因不会更改分辨率,尽管我认为至少在 Qt4 window 如果我使用 show() 而不是 savefig()。 请注意,无论您在 plt.figure(dpi=).

中设置什么,保存的数字中的分辨率都是相同的

我不知道该系统如何工作,而且我的理解也有限。非常欢迎任何帮助。

提前致谢。

首先,当您另存为 .pdf 时,您隐含地使用了 pdf 后端,即使您可能在选项中指定了其他后端。这意味着您的图像以矢量格式保存,因此 dpi 毫无意义。在任何分辨率下,如果我在一个像样的查看器中加载您的 PDF(我使用 inkscape,其他可用),您可以清楚地看到条纹 - 实际上我发现如果您将每隔一行设置为零,则更容易观察。所有生成的 PDF 都包含复制条纹的完整信息,因此几乎完全相同。当您指定 figsize=(45, 10) 时,所有生成的 PDF 都建议显示尺寸为 45 英寸 x 10 英寸。

如果我将 png 指定为图像类型,我会看到基于 dpi 参数的文件大小有所不同,我认为这正是您所期望的。如果您查看 100 dpi 图像,它有 4500000 个像素,200 dpi 图像有 18000000 个像素(4 倍),300 dpi 图像有 40500000(9 倍)。您会注意到 4500000 == 1500 x 3000,即原始数组的每个成员一个像素。因此,较大的 dpi 设置实际上不会让您获得任何进一步的定义 - 相反,您的条纹分别为 2 或 3 像素宽而不是 1.

认为 你想要做的是有效地绘制每列 10 次,这样你就可以得到一个 1500 x 30000 像素的图像。为此,使用您自己的所有代码,您可以使用 np.repeat 执行如下操作:

import numpy as np
import matplotlib.pyplot as plt
import matplotlib.colors

R, C = 1500, 3000
DATA = np.random.random((R, C))
DATA[::2, :] = 0  # make every other line plain white
Yi, Xi = 1, 10 # increment
DATA = np.repeat(DATA, Xi, axis=1)
DATA = np.repeat(DATA, Yi)

CMP = 'seismic'
ImageFormat ='pdf'
Name = 'Image'


DataRange = (np.absolute(DATA)).max() # I want my data centred on 0
EXTENT = [0, Xi*C, 0 ,Yi*R]
NORM = matplotlib.colors.Normalize(vmin =-DataRange, vmax= DataRange, clip =True)

for i in range(1,4):
    Fig=plt.figure(figsize=(45, 10), dpi = 100*i, tight_layout=True)
    Fig.suptitle(Name+str(i)+'00DPI')
    ax = Fig.add_subplot(1, 1, 1)
    Plot = ax.imshow(DATA, cmap=plt.get_cmap(CMP), norm = NORM, extent = EXTENT, aspect = 1, interpolation='none') 
    ax.set_xlabel('metres')
    ax.set_ylabel('metres')
    Fig.savefig(Name+str(i)+'00DPI.'+ImageFormat,  format = ImageFormat, dpi = Fig.dpi)
plt.close()

警告: 这是一个内存密集型解决方案 - 可能有更好的方法。如果你不需要pdf的矢量图输出,你可以把你的ImageFormat变量改成png


令我印象深刻的是,您可能关心的另一件事是为图片提供适当的宽高比(即宽是高的 20 倍)。这你已经在做了。因此,如果您查看 pdf 中每个像素的表示,它们都是矩形的(宽度是高度的 10 倍),而不是正方形。

运行 你的例子,缩放后 matplotlib 中的一切看起来都很好:无论分辨率如何,结果都是一样的,我看到每个轴单位一个像素。 此外,尝试使用较小的数组、pdf(或其他格式)效果很好。

这是我的解释:当你设置图形dpi时,你设置的是整个图形的dpi(不仅仅是数据区域)。在我的系统上,这导致绘图区域垂直占据整个图形的大约 20%。如果你设置 300 dpi 和 10 的高度,你得到垂直数据轴总共 300x10x0.2=600 像素,这不足以代表 1500 个点,这向我解释了为什么输出必须重新采样。请注意,减小宽度有时会偶然起作用,因为它会改变数据图所占图形的比例。

然后你必须增加dpi并设置interpolation='none'(分辨率设置是否完美应该无关紧要,但如果它足够接近就很重要)。 您还可以调整绘图位置和大小以占据图形的较大部分,但回到最佳分辨率设置,理想情况下您希望轴上的像素数量是数据点的倍数,否则一些必须进行某种插值(想想如何在三个像素上绘制两个点,反之亦然)。

我不知道以下是否是最好的方法,matplotlib 中可能有更合适的方法和属性,但我会尝试这样计算最佳 dpi:

vsize=ax.get_position().size[1]  #fraction of figure occupied by axes
axesdpi= int((Fig.get_size_inches()[1]*vsize)/R)  #(or Yi*R according to what you want to do)

然后你的代码(减少到第一个循环)变成:

import numpy as np
import matplotlib.pyplot as plt
import matplotlib.colors

R, C = 1500, 3000
DATA = np.random.random((R, C))
DATA[::2, :] *= -1  # make every other line negative
Yi, Xi = 1, 10 # increment
CMP = 'seismic'
ImageFormat ='pdf'
Name = 'Image'


DataRange = (np.absolute(DATA)).max() # I want my data centred on 0
EXTENT = [0, Xi*C, 0 ,Yi*R]
NORM = matplotlib.colors.Normalize(vmin =-DataRange, vmax= DataRange, clip =True)

for i in (1,):
    print i 
    Fig=plt.figure(figsize=(45, 10), dpi = 100*i, tight_layout=True)
    Fig.suptitle(Name+str(i)+'00DPI')
    ax = Fig.add_subplot(1, 1, 1)
    Plot = ax.imshow(DATA, cmap=plt.get_cmap(CMP), norm = NORM, extent = EXTENT, aspect = 1, interpolation='none') 
    ax.set_xlabel('metres')
    ax.set_ylabel('metres')
    vsize=ax.get_position().size[1]  #fraction of figure occupied by axes
    axesdpi= int((Fig.get_size_inches()[1]*vsize)/R)  #(or Yi*R according to what you want to do)
    Fig.savefig(Name+str(axesdpi)+'DPI.'+ImageFormat,  format = ImageFormat, dpi = axesdpi)
    #plt.close()

这对我来说很有效。