詳解Python中plt.savefig的使用

一、簡介

Python中的matplotlib庫可用於繪製數據圖表，而plt.savefig則是用於將繪製的圖表保存為圖片文件的函數。在繪製圖表時，plt.savefig可以幫助用戶將數據圖表保存為多種常用格式的圖片文件，簡化用戶的工作。本文將介紹plt.savefig的使用方法，包括文件格式選擇、參數設置等。

二、保存為圖片

plt.savefig將當前的繪圖窗口保存為一個指定格式的文件，支持多種格式，包括PNG、PDF、SVG、JPEG和TIFF等。該函數的基本調用方法如下：

plt.savefig(filename, dpi=None, facecolor='w', edgecolor='w',
        orientation='portrait', papertype=None, format=None,
        transparent=False, bbox_inches=None, pad_inches=0.1,
        frameon=None, metadata=None)

其中filename是必須參數，表示保存的文件名及路徑，格式可為字符串或者路徑對象。圖像格式由文件名的擴展名確定，例如.png表示保存為PNG格式文件。

三、設置圖像大小和DPI

可以通過設置dpi參數，調節輸出圖像的清晰度和大小。如果不指定dpi，那麼將採用默認的dpi值。dpi的默認值為100。

可以通過設置figsize參數，調節圖像的大小。例如，代碼中figsize=(8, 6)表示將圖像寬度設置為8英寸，高度設置為6英寸。

import matplotlib.pyplot as plt
import numpy as np

x = np.linspace(-np.pi, np.pi, 256,endpoint=True)
y = np.sin(x)
plt.plot(x,y)
plt.xlim(-4.0,4.0)
plt.ylim(-1.0,1.0)

plt.savefig("sin_wave_dpi.png",dpi=200)#dpi值為200
plt.savefig("sin_wave_size.png", figsize=(8, 6))#設置圖像大小

四、設置圖片邊界和顏色

可以通過通過設置bbox_inches參數，設置保存圖像的邊界尺寸，以去掉圖片周圍多餘的空白部分。其中參數tight可以使圖像更加緊湊。

可以通過設置facecolor參數，設置圖像的背景顏色。也可以通過設置edgecolor參數，設置圖像的邊框顏色。

plt.plot(x,y)
plt.xlim(-4.0,4.0)
plt.ylim(-1.0,1.0)

plt.savefig("sin_wave_bbox.png", bbox_inches='tight',pad_inches=0)#去掉多餘空白部分
plt.savefig("sin_wave_color.png", facecolor='g',edgecolor='r')#設置背景和邊框顏色

五、設置圖片的透明度

可以通過設置transparent參數，設置保存圖像的透明度，使得圖像更加半透明。

plt.plot(x,y)
plt.xlim(-4.0,4.0)
plt.ylim(-1.0,1.0)

plt.savefig("sin_wave_transparent.png", transparent=True)#設置透明度

六、保存為PDF格式

當需要將繪製的數據圖表保存為PDF格式文件時，可使用如下方法。

plt.plot(x,y)
plt.xlim(-4.0,4.0)
plt.ylim(-1.0,1.0)

plt.savefig("sin_wave.pdf", format='pdf', dpi=1000)#保存為PDF

七、保存為SVG格式

當需要將繪製的數據圖表保存為SVG格式文件時，可使用如下方法。

plt.plot(x,y)
plt.xlim(-4.0,4.0)
plt.ylim(-1.0,1.0)

plt.savefig("sin_wave.svg", format='svg', dpi=1000)#保存為SVG

八、總結

本文的重點是介紹plt.savefig函數的使用方法和注意事項。在使用plt.savefig函數時，需要注意保存的文件名和文件格式，以及設置dpi、 figsize等參數，以得到最佳的輸出結果。