Matplotlib绘图:删除轴,图例和空格

Question

我是Python和Matplotlib的新手,我想简单地将colormap应用于图像并编写生成的图像,而不使用轴,标签,标题或通常由matplotlib自动添加的任何内容.这是我做的:

def make_image(inputname,outputname):
    data = mpimg.imread(inputname)[:,:,0]
    fig = plt.imshow(data)
    fig.set_cmap('hot')
    fig.axes.get_xaxis().set_visible(False)
    fig.axes.get_yaxis().set_visible(False)
    plt.savefig(outputname)

它成功地移除了图形的轴,但保存的图形呈现白色填充和实际图像周围的框架.如何删除它们(至少是白色填充)？谢谢

Answer 1

我认为该命令axis('off')比单独更改每个轴和边界更简洁地处理其中一个问题.然而,它仍然留下边界周围的白色空间.添加bbox_inches='tight'到savefig命令几乎可以到达那里,您可以在下面的示例中看到剩下的空白区域要小得多,但仍然存在.

from numpy import random
import matplotlib.pyplot as plt

data = random.random((5,5))
img = plt.imshow(data, interpolation='nearest')
img.set_cmap('hot')
plt.axis('off')
plt.savefig("test.png", bbox_inches='tight')

Answer 2

我从matehat那里学到了这个技巧,在这里:

import matplotlib.pyplot as plt
import numpy as np

def make_image(data, outputname, size=(1, 1), dpi=80):
    fig = plt.figure()
    fig.set_size_inches(size)
    ax = plt.Axes(fig, [0., 0., 1., 1.])
    ax.set_axis_off()
    fig.add_axes(ax)
    plt.set_cmap('hot')
    ax.imshow(data, aspect='equal')
    plt.savefig(outputname, dpi=dpi)

# data = mpimg.imread(inputname)[:,:,0]
data = np.arange(1,10).reshape((3, 3))

make_image(data, '/tmp/out.png')

产量

Answer 3

最简单的解决方案:

我简单地将问题中描述的方法与Hooked的答案中的方法结合起来.

fig = plt.imshow(my_data)
plt.axis('off')
fig.axes.get_xaxis().set_visible(False)
fig.axes.get_yaxis().set_visible(False)
plt.savefig('pict.png', bbox_inches='tight', pad_inches = 0)

在此代码之后,没有空格,也没有框架.

Answer 4

还没有人提到imsave,这使得它成为一个单行:

import matplotlib.pyplot as plt
import numpy as np

data = np.arange(10000).reshape((100, 100))
plt.imsave("/tmp/foo.png", data, format="png", cmap="hot")

它直接存储图像,即不添加任何轴或边框/填充.

Answer 5

plt.axis('off')

plt.savefig('example.png',bbox_inches='tight',pad_inches = 0)

让我得到无边框图像。

Answer 6

我发现这一切都记录在案......

https://matplotlib.org/3.1.1/api/_as_gen/matplotlib.axes.Axes.axis.html#matplotlib.axes.Axes.axis

我的代码…… “bcK”是一个 512x512 的图像

plt.figure()
plt.imshow(bck)
plt.axis("off")   # turns off axes
plt.axis("tight")  # gets rid of white border
plt.axis("image")  # square up the image instead of filling the "figure" space
plt.show()

Answer 7

您还可以为参数指定图形的范围bbox_inches.这将摆脱图周围的白色填充.

def make_image(inputname,outputname):
    data = mpimg.imread(inputname)[:,:,0]
    fig = plt.imshow(data)
    fig.set_cmap('hot')
    ax = fig.gca()
    ax.set_axis_off()
    ax.autoscale(False)
    extent = ax.get_window_extent().transformed(plt.gcf().dpi_scale_trans.inverted())
    plt.savefig(outputname, bbox_inches=extent)

Answer 8

这应该删除所有填充和边框：

from matplotlib import pyplot as plt

fig = plt.figure()
fig.patch.set_visible(False)

ax = fig.add_subplot(111)

plt.axis('off')
plt.imshow(data)

extent = ax.get_window_extent().transformed(fig.dpi_scale_trans.inverted())
plt.savefig("../images/test.png", bbox_inches=extent)

Answer 9

我喜欢ubuntu 的答案，但它没有明确显示如何设置开箱即用的非方形图像的大小，所以我修改了它以便于复制粘贴：

import matplotlib.pyplot as plt
import matplotlib.image as mpimg
import numpy as np

def save_image_fix_dpi(data, dpi=100):
    shape=np.shape(data)[0:2][::-1]
    size = [float(i)/dpi for i in shape]

    fig = plt.figure()
    fig.set_size_inches(size)
    ax = plt.Axes(fig,[0,0,1,1])
    ax.set_axis_off()
    fig.add_axes(ax)
    ax.imshow(data)
    fig.savefig('out.png', dpi=dpi)
    plt.show()

如果保留 Pixel_size/dpi=size，无论您选择什么 dpi，保存无边框图像都很容易。

data = mpimg.imread('test.png')
save_image_fix_dpi(data, dpi=100)

然而显示却很诡异。如果您选择较小的 dpi，则图像尺寸可能会大于屏幕，并且在显示过程中会出现边框。不过，这并不影响储蓄。

因此对于

save_image_fix_dpi(data, dpi=20)

显示变为有边框（但保存有效）：

Answer 10

赞成的答案不再有效。要使其正常工作，您需要手动添加设置为 [0, 0, 1, 1] 的轴，或删除图下的补丁。

import matplotlib.pyplot as plt

fig = plt.figure(figsize=(5, 5), dpi=20)
ax = plt.Axes(fig, [0., 0., 1., 1.])
fig.add_axes(ax)
plt.imshow([[0, 1], [0.5, 0]], interpolation="nearest")
plt.axis('off')                                # same as: ax.set_axis_off()

plt.savefig("test.png")

或者，您可以删除补丁。您不需要添加子图来删除填充。这是从下面 Vlady 的回答中简化的

fig = plt.figure(figsize=(5, 5))
fig.patch.set_visible(False)                   # turn off the patch

plt.imshow([[0, 1], [0.5, 0]], interpolation="nearest")
plt.axis('off')

plt.savefig("test.png", cmap='hot')

这是3.0.3在 2019/06/19 上测试的版本。图片见下图：

更简单的事情是使用pyplot.imsave. 详情请看下面luator的回答