在 Python 中保存绘图

当我们使用 Python 进行数据分析和可视化时，我们通常会使用一些绘图库如 Matplotlib 或 Seaborn 来可视化我们的数据。当我们生成了一个绘图，我们可能会想将其保存到本地磁盘或内存中以备以后使用或共享。

在 Python 中，我们可以使用 Matplotlib 库的 savefig() 方法来保存我们的图形。让我们看看在 Python 中如何保存绘图。

安装 Matplotlib 库

如果我们还没有安装 Matplotlib 库，我们可以使用以下命令在我们的终端或命令提示符中安装它：

pip install matplotlib

配置 Matplotlib

在保存图形之前，我们需要一些必要的配置。我们需要导入 Matplotlib 库并配置它以使用我们所选的绘图引擎。在此期间，我们可以选择设置默认字体大小并选择保存图形的格式。

import matplotlib.pyplot as plt

# 嵌入到 notebook 中
%matplotlib inline

# 如果你是在 cmd 中使用 matplotlib，可以使用下面这行配置。
# %config InlineBackend.figure_format = 'svg'

# 设置字体大小
plt.rcParams['font.size'] = 12

# 图形的格式可以是 'png', 'pdf', 'svg', 'ps' 和 'eps' 等
plt.rcParams['savefig.format'] = 'png'

生成一个简单的绘图

接下来，我们将生成一个简单图形，以便我们可以演示如何保存它。我们将使用 sin() 函数生成一个函数图形，并使用 plot() 方法将其绘制。最后，我们使用 show() 方法显示图形。

import numpy as np

x = np.linspace(0, 2 * np.pi, 100)
y = np.sin(x)

plt.plot(x, y)
plt.show()

这将生成以下图形：

保存绘图

使用 savefig() 方法可以将我们的绘图保存到磁盘中。我们可以选择指定要保存的文件名和路径。如果我们没有指定路径，则默认将保存到当前工作目录中。

plt.plot(x, y)
plt.savefig('sin_wave.png')

这将把我们的图形保存到一个名为 sin_wave.png 的文件中。我们可以使用以下代码片段来随时查看所保存的图形。

from IPython.display import Image

Image(filename='sin_wave.png')

这将在我们的 Python 环境中显示保存的图形。

如果我们不需要真正地保存图形，并只是想将其保留在内存中，则可以使用 BytesIO 类，该类允许我们将二进制数据保存到内存中。以下代码展示了如何使用 BytesIO 类在内存中保存我们的图形。

from io import BytesIO

buffer = BytesIO()
plt.savefig(buffer, format='png')
image_content = buffer.getvalue()

# 现在我们可以使用 image_content 了，例如，我们可以将其写入文件或通过网络共享。

结论

在 Python 中保存绘图非常简单。我们只需要配置 Matplotlib 库并使用 savefig() 方法即可将绘图保存到本地磁盘或内存中。这使得我们可以随时保存和共享我们的图形。