在 matplotlib 中轻松去除坐标轴、图例和白边

在 matplotlib 中去除坐标轴、图例和白边

引言

matplotlib 是一个流行的 Python 库，用于创建可视化，例如图像和图形。虽然它提供了强大的工具来定制这些可视化，但有时您可能希望删除不需要的元素，例如坐标轴、图例和白边。这篇文章将提供分步指南，帮助您解决这些问题。

移除坐标轴

坐标轴通常用于指示图表的轴向值。如果您不需要这些信息，则可以使用 plt.axis('off') 将其删除。这将隐藏 x 和 y 轴以及相关的刻度和标签。

移除图例

图例通常用于解释图表中的不同颜色或符号。要删除图例，您可以使用 plt.legend().remove()。这将从绘图中删除所有图例项。

移除白边

白边通常是由于 matplotlib 的绘图区域大于图像本身造成的。有两种方法可以解决这个问题：

方法 1：使用 tight_layout()

plt.tight_layout() 会自动调整绘图区域的大小，使其适合图像。此方法通常有效，但如果图像周围有其他元素（例如标题或子图），则可能不够用。

方法 2：使用 subplots_adjust()

plt.subplots_adjust() 允许您手动调整绘图区域的边距。您可以使用以下代码设置所有边距为零，从而去除白边：

plt.subplots_adjust(left=0, right=1, bottom=0, top=1)

示例代码

以下示例代码将移除坐标轴、图例和白边：

import matplotlib.pyplot as plt
import numpy as np

# 加载图像
image = plt.imread('image.jpg')

# 应用颜色图
plt.imshow(image)
plt.set_cmap('hot')

# 移除坐标轴
plt.axis('off')

# 移除图例
plt.legend().remove()

# 移除白边（方法 1）
# plt.tight_layout()

# 移除白边（方法 2）
plt.subplots_adjust(left=0, right=1, bottom=0, top=1)

# 保存图像
plt.savefig('output_image.jpg', bbox_inches='tight', pad_inches=0)

结论

通过遵循这些步骤，您可以轻松地从 matplotlib 中的图像中删除坐标轴、图例和白边。这对于创建干净且专业的可视化非常有用。

常见问题解答

• 为什么我仍然看到白边？
  • 确保您使用了正确的 bbox_inches 和 pad_inches 参数来保存图像。
  • 尝试使用 plt.margins(0) 进一步调整绘图区域的边距。
• 如何移除图像周围的框架？
  • 使用 plt.gca().set_frame_on(False) 即可移除图像周围的框架。
• 如何在不移除 x 轴的情况下移除 y 轴？
  • 使用 plt.gca().get_yaxis().set_visible(False) 即可隐藏 y 轴。
• 如何设置自定义的坐标轴标签？
  • 使用 plt.gca().set_xlabel() 和 plt.gca().set_ylabel() 可以设置自定义的坐标轴标签。
• 如何在matplotlib中改变坐标轴的字体大小？
  • 使用 plt.gca().tick_params(axis='both', which='major', labelsize=12) 即可改变坐标轴的字体大小。