如何完全消除散点图周围的空白

2022-05-24 python matplotlib scatter-plot plot imshow

问题描述

我正在尝试在没有任何空白的图像上绘制散点图。

如果我只按如下方式绘制图像，则没有空白：

fig = plt.imshow(im,alpha=alpha,extent=(0,1,1,0))
plt.axis('off')
fig.axes.axis('tight')
fig.axes.get_xaxis().set_visible(False)
fig.axes.get_yaxis().set_visible(False)

但当我在图像上添加散点图时，如下所示：

fig = plt.scatter(sx, sy,c="gray",s=4,linewidths=.2,alpha=.5)
fig.axes.axis('tight')
fig.axes.get_xaxis().set_visible(False)
fig.axes.get_yaxis().set_visible(False)

此时，通过使用以下savefig命令，将在图像周围添加空白：

plt.savefig(im_filename,format="png",bbox_inches='tight',pad_inches=0)

有什么办法可以确定删除空格吗？

解决方案

通过切换到mpl面向对象的样式，您可以在同一轴上同时绘制图像和散点图，因此只需使用ax.imshow和ax.scatter设置一次空白。

在下面的示例中，我使用subplots_adjust删除了轴周围的空格，并使用ax.axis('tight')设置了数据范围的轴限制。

import matplotlib.pyplot as plt
import numpy as np

# Load an image
im = plt.imread('stinkbug.png')

# Set the alpha
alpha = 0.5

# Some random scatterpoint data
sx = np.random.rand(100)
sy = np.random.rand(100)

# Creare your figure and axes
fig,ax = plt.subplots(1)

# Set whitespace to 0
fig.subplots_adjust(left=0,right=1,bottom=0,top=1)

# Display the image
ax.imshow(im,alpha=alpha,extent=(0,1,1,0))

# Turn off axes and set axes limits
ax.axis('tight')
ax.axis('off')

# Plot the scatter points
ax.scatter(sx, sy,c="gray",s=4,linewidths=.2,alpha=.5)

plt.show()

相关文章