python - 使用 python matplotlib 仅将轴区域导出到位图

Question

对于科学会议，论文的文件大小通常是有限的。我喜欢将我的图包含为 pdf，因此文本和线条保持清晰。但是，当我创建带有大量数据的假彩色图或散点图时，导出的 pdf 很容易变得比整篇论文允许的更大。

有没有一种方法可以仅将轴区域导出到位图，以便稍后将其包含在矢量轴中？（或者有没有更好的方法来获取一些嵌入为位图的元素的 pdf？）

我希望有人可以在这里帮助我。由于这是我的第一篇文章，因此感谢您对如何改进我的问题的评论。

Answer 1

You can tell individual Artists to be exported as rastered in vector output:

img = plt.imshow(...)
img.set_rasterized(True)

(doc)

Answer 2

我意识到这种解决方法适用于大型散点图。它一点也不优雅，但它为我解决了这个问题。非常欢迎关于更优雅解决方案的建议；）

    # -*- coding: utf-8 -*-
"""
Created on Sat Jun  1 17:21:53 2013

@author: roel
"""

import pylab as pl
from matplotlib._png import read_png


def bitmappify(ax, dpi=None):
    fig = ax.figure
    # safe plot without axes
    ax.set_axis_off()
    fig.savefig('bitmap.png', dpi=dpi, transparent=True)
    ax.set_axis_on()

    # remeber geometry
    xl = ax.get_xlim()
    yl = ax.get_ylim()
    xb = ax.bbox._bbox.corners()[:,0]
    xb = (min(xb), max(xb))
    yb = ax.bbox._bbox.corners()[:,1]
    yb = (min(yb), max(yb))

    # compute coordinates to place bitmap image later
    xb = (- xb[0] / (xb[1] - xb[0]),
        (1 - xb[0]) / (xb[1] - xb[0]))
    xb = (xb[0] * (xl[1] - xl[0]) + xl[0],
        xb[1] * (xl[1] - xl[0]) + xl[0])
    yb = (- yb[0] / (yb[1] - yb[0]),
        (1 - yb[0]) / (yb[1] - yb[0]))
    yb = (yb[0] * (yl[1] - yl[0]) + yl[0],
        yb[1] * (yl[1] - yl[0]) + yl[0])

    # replace the dots by the bitmap
    del ax.collections[:]
    del ax.lines[:]
    ax.imshow(read_png('bitmap.png'), origin='upper',
             aspect= 'auto', extent=(xb[0], xb[1], yb[0], yb[1]))

    # reset view
    ax.set_xlim(xl)
    ax.set_ylim(yl)

# create a plot
f, a = pl.subplots(1,1)
n = 1e4
a.scatter(pl.random(n)*2+6, pl.random(n)*3-12,
          c=pl.random(n), s=100*pl.random(n))

# safe as large pdf file for comparison
f.savefig('vector.pdf') # gives a large file: 3.8 MB

bitmappify(a, 144)

# save as smaller pdf
f.savefig('hybrid.pdf', dpi=144) # reasonably sized file: 0.5 MB