如何将 matplotlib 图形保存为 png - Python

在数据可视化过程中，matplotlib 是一个非常强大的 Python 库。一旦你创建了一个美观的图形，你可能想把它保存下来用于报告、展示或分享。你可以使用 matplotlib 内置的 savefig() 函数把图形保存为多种不同的文件格式，例如 PNG、JPG、PDF 和 SVG 等等。本文将向你展示如何将 matplotlib 图形保存为 PNG 格式。

步骤 1 - 导入必要的库和创建图形

首先，你需要导入必要的库（matplotlib 和 numpy），并创建一个简单的图形，以便我们可以将其保存为 PNG 格式。

import matplotlib.pyplot as plt
import numpy as np

# 创建一些随机数据
x = np.random.randint(0, 10, size=10)
y = np.random.randint(0, 10, size=10)

# 创建散点图
plt.scatter(x, y)
plt.title("随机散点图")
plt.xlabel("X 轴")
plt.ylabel("Y 轴")
plt.show()

步骤 2 - 保存图形为 PNG 格式

现在，我们已经创建了图形，可以使用 savefig() 函数将其保存为 PNG 格式。

# 保存图形为 PNG 格式
plt.savefig("scatter.png")

# 显示图形
plt.show()

在运行代码后，你会发现电脑当前目录下多了一个 scatter.png 文件，这就是我们刚才保存的 PNG 格式文件。

完整代码

import matplotlib.pyplot as plt
import numpy as np

# 创建一些随机数据
x = np.random.randint(0, 10, size=10)
y = np.random.randint(0, 10, size=10)

# 创建散点图
plt.scatter(x, y)
plt.title("随机散点图")
plt.xlabel("X 轴")
plt.ylabel("Y 轴")

# 保存图形为 PNG 格式
plt.savefig("scatter.png")

# 显示图形
plt.show()

总结

本文向你展示了如何将 matplotlib 图形保存为 PNG 格式文件。你可以根据需要将其保存为其他格式的文件，只需要使用 savefig() 函数并相应地更改文件扩展名即可。当你需要将你的数据可视化结果在论文、报告或网站中展示时，这将会非常有用。