plt.savefig用法详解

小蓝 • 2024-11-12 00:54 • 编程

一、基本用法

plt.savefig是matplotlib中用于保存图形的函数。基本用法如下：

import matplotlib.pyplot as plt
plt.plot([0,1,2,3,4],[0,1,4,9,16])
plt.savefig('myplot.png')

这段代码会绘制一条曲线，并将图形保存在当前工作目录中的myplot.png文件中。

plt.savefig的第一个参数是保存的文件名，可以是带有文件路径的完整路径。如：

plt.savefig('/home/user/myplot.png')

除了png格式，还可以保存为pdf、ps、eps、svg等格式。要保存为pdf格式，只需要将文件名的扩展名改成pdf即可。

二、调整图形大小

如果默认的图形大小不满意，可以通过figsize参数来调整。该参数的单位是英寸，可以是一个二元组或列表。如：

plt.plot([0,1,2,3,4],[0,1,4,9,16])
plt.savefig('myplot.png', figsize=(6,4))

这段代码会将图形宽度设为6英寸，高度设为4英寸。

三、调整分辨率

如果保存的图形在不同的设备上需要显示，可能需要调整分辨率。可以使用dpi参数来指定分辨率。如：

plt.plot([0,1,2,3,4],[0,1,4,9,16])
plt.savefig('myplot.png', dpi=300)

这段代码会将分辨率设为300，使得图形在高分辨率屏幕上显示效果更好。

四、保存为多个子图

如果要保存为多个子图，可以使用plt.subplots函数来创建。如：

import matplotlib.pyplot as plt
fig, ax = plt.subplots(2, 2)
ax[0, 0].plot([0, 1, 2, 3, 4], [0, 1, 4, 9, 16])
ax[0, 1].scatter([0, 1, 2, 3, 4], [0, 1, 4, 9, 16])
ax[1, 0].bar([0, 1, 2, 3, 4], [0, 1, 4, 9, 16])
ax[1, 1].pie([0, 1, 4, 9, 16])
plt.savefig('myplots.png')

这段代码会将4个子图保存在同一个文件中。

五、调整边缘空白

默认情况下，保存的图形会有一定的边缘空白。可以通过调整plt.tight_layout函数的参数来去除边缘空白。如：

import matplotlib.pyplot as plt
fig, ax = plt.subplots(2, 2)
ax[0, 0].plot([0, 1, 2, 3, 4], [0, 1, 4, 9, 16])
ax[0, 1].scatter([0, 1, 2, 3, 4], [0, 1, 4, 9, 16])
ax[1, 0].bar([0, 1, 2, 3, 4], [0, 1, 4, 9, 16])
ax[1, 1].pie([0, 1, 4, 9, 16])
plt.savefig('myplots.png')
plt.tight_layout(pad=0)

这段代码会去除边缘空白，使得4个子图紧凑排列。

六、保存为透明背景

如果要将绘制的图形叠加到其他图形之上，可以将背景设置为透明。可以使用transparent参数来指定透明度。如：

import matplotlib.pyplot as plt
plt.plot([0, 1, 2, 3, 4], [0, 1, 4, 9, 16])
plt.savefig('myplot.png', transparent=True)

这段代码会将保存的图形背景设为透明。

七、保存为灰度图

如果要将保存的图形转换为灰度图，可以将cmap参数设置为’gray’。如：

import matplotlib.pyplot as plt
plt.plot([0, 1, 2, 3, 4], [0, 1, 4, 9, 16])
plt.savefig('myplot.png', cmap='gray')

这段代码会将保存的图形转换为灰度图。

原创文章，作者：小蓝，如若转载，请注明出处：https://www.506064.com/n/151962.html