在matplotlib中保存子图

uqdfh47h  于 2023-10-24  发布在  其他
关注(0)|答案(2)|浏览(157)

是否可以在matplotlib图中保存(到png)单个子图?

import pyplot.matplotlib as plt
ax1 = plt.subplot(121)
ax2 = plt.subplot(122)
ax1.plot([1,2,3],[4,5,6])    
ax2.plot([3,4,5],[7,8,9])

是否可以将两个子图分别保存到不同的文件中,或者至少将它们单独复制到一个新的图形中以保存它们?
我在RHEL 5上使用matplotlib 1.0.0版本。

ttcibm8c

ttcibm8c1#

虽然@Eli说得很对,通常没有太多的必要这样做,但这是可能的。savefig接受一个bbox_inches参数,该参数可用于选择性地将图形的一部分保存到图像中。
下面是一个简单的例子:

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

# Make an example plot with two subplots...
fig = plt.figure()
ax1 = fig.add_subplot(2,1,1)
ax1.plot(range(10), 'b-')

ax2 = fig.add_subplot(2,1,2)
ax2.plot(range(20), 'r^')

# Save the full figure...
fig.savefig('full_figure.png')

# Save just the portion _inside_ the second axis's boundaries
extent = ax2.get_window_extent().transformed(fig.dpi_scale_trans.inverted())
fig.savefig('ax2_figure.png', bbox_inches=extent)

# Pad the saved area by 10% in the x-direction and 20% in the y-direction
fig.savefig('ax2_figure_expanded.png', bbox_inches=extent.expanded(1.1, 1.2))

图片:x1c 0d1x
第二个子图内的面积:

第二子图周围的面积在x方向上填充10%,在y方向上填充20%:

oaxa6hgo

oaxa6hgo2#

在@Joe 3年后从here得到的答案中应用full_extent()函数,您可以确切地获得OP正在寻找的内容。或者,您可以使用Axes.get_tightbbox(),它提供了一个更紧密的边界框。

import matplotlib.pyplot as plt
import matplotlib as mpl
import numpy as np
from matplotlib.transforms import Bbox

def full_extent(ax, pad=0.0):
    """Get the full extent of an axes, including axes labels, tick labels, and
    titles."""
    # For text objects, we need to draw the figure first, otherwise the extents
    # are undefined.
    ax.figure.canvas.draw()
    items = ax.get_xticklabels() + ax.get_yticklabels() 
#    items += [ax, ax.title, ax.xaxis.label, ax.yaxis.label]
    items += [ax, ax.title]
    bbox = Bbox.union([item.get_window_extent() for item in items])

    return bbox.expanded(1.0 + pad, 1.0 + pad)

# Make an example plot with two subplots...
fig = plt.figure()
ax1 = fig.add_subplot(2,1,1)
ax1.plot(range(10), 'b-')

ax2 = fig.add_subplot(2,1,2)
ax2.plot(range(20), 'r^')

# Save the full figure...
fig.savefig('full_figure.png')

# Save just the portion _inside_ the second axis's boundaries
extent = full_extent(ax2).transformed(fig.dpi_scale_trans.inverted())
# Alternatively,
# extent = ax.get_tightbbox(fig.canvas.renderer).transformed(fig.dpi_scale_trans.inverted())
fig.savefig('ax2_figure.png', bbox_inches=extent)

我想发张照片,但我没有声望值

相关问题