Plot matplotlib: menghilangkan sumbu, legenda dan spasi putih

Saya baru mengenal Python dan Matplotlib, saya hanya ingin menerapkan colormap ke gambar dan menulis gambar yang dihasilkan, tanpa menggunakan sumbu, label, judul atau apa pun yang biasanya secara otomatis ditambahkan oleh matplotlib. Inilah yang saya lakukan:

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)

Ini berhasil menghilangkan sumbu gambar, tetapi gambar yang disimpan menyajikan padding putih dan bingkai di sekitar gambar yang sebenarnya. Bagaimana saya bisa menghapusnya (setidaknya lapisan putih)? Terima kasih

Saya berpikir bahwa perintah axis('off')menangani salah satu masalah lebih ringkas daripada mengubah setiap sumbu dan perbatasan secara terpisah. Namun masih meninggalkan ruang putih di sekitar perbatasan. Menambahkan bbox_inches='tight'ke savefigperintah hampir membuat Anda di sana, Anda dapat melihat pada contoh di bawah ini bahwa ruang putih yang tersisa jauh lebih kecil, tetapi masih ada.

Perhatikan bahwa versi yang lebih baru dari matplotlib mungkin memerlukan bbox_inches=0alih-alih string 'tight'(via @episodeyang dan @kadrach)

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')

Saya belajar trik ini dari matehat, di sini :

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')

hasil panen

Kemungkinan solusi paling sederhana:

Saya hanya menggabungkan metode yang dijelaskan dalam pertanyaan dan metode dari jawaban oleh 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)

Setelah kode ini tidak ada spasi putih dan tidak ada bingkai.

Tidak ada yang disebutkan imsave, yang membuat ini satu kalimat:

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")

Langsung menyimpan gambar apa adanya, yaitu tidak menambahkan sumbu atau perbatasan / padding.

Anda juga bisa menentukan tingkat gambar hingga bbox_inchesargumen. Ini akan menghilangkan lapisan putih di sekitar sosok itu.

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)

Ini harus menghapus semua bantalan dan batas:

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)

Saya menemukan bahwa itu semua didokumentasikan ...

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

Kode saya .... "bcK" adalah gambar 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()

Jawaban terunggah tidak berfungsi lagi. Untuk membuatnya bekerja, Anda perlu secara manual menambahkan sumbu yang diatur ke [0, 0, 1, 1], atau menghapus patch di bawah gambar.

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")

Atau, Anda bisa menghapus tambalan. Anda tidak perlu menambahkan subplot untuk menghapus paddings. Ini disederhanakan dari jawaban Vlady di bawah ini

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')

Ini diuji dengan versi 3.0.3pada 2019/06/19. Gambar lihat di bawah:

Hal yang lebih sederhana untuk dilakukan adalah menggunakan pyplot.imsave. Untuk detailnya, lihat jawaban luator di bawah

Saya menyukai jawaban ubuntu , tetapi itu tidak menunjukkan secara eksplisit bagaimana mengatur ukuran untuk gambar non-kotak out-of-the-box, jadi saya memodifikasinya untuk mudah salin-tempel:

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()

Menyimpan gambar tanpa batas itu mudah, apa pun dpi yang Anda pilih jika pixel_size / dpi = ukuran disimpan.

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

Namun tampilan itu seram. Jika Anda memilih dpi kecil, ukuran gambar Anda bisa lebih besar dari layar Anda dan Anda mendapatkan batas selama tampilan. Namun demikian, ini tidak mempengaruhi tabungan.

Sehingga untuk

save_image_fix_dpi(data, dpi=20)

Layar menjadi berbatasan (tetapi menyimpan karya):

Pertama, untuk format gambar tertentu (yaitu TIFF ) Anda sebenarnya dapat menyimpan colormap di header dan sebagian besar pemirsa akan menampilkan data Anda dengan colormap.

Untuk menyimpan matplotlibgambar yang sebenarnya , yang dapat berguna untuk menambahkan anotasi atau data lain ke gambar, saya telah menggunakan solusi berikut:

fig, ax = plt.subplots(figsize=inches)
ax.matshow(data)  # or you can use also imshow
# add annotations or anything else
# The code below essentially moves your plot so that the upper
# left hand corner coincides with the upper left hand corner
# of the artist
fig.subplots_adjust(left=0, right=1, top=1, bottom=0, wspace=0, hspace=0)
# now generate a Bbox instance that is the same size as your
# single axis size (this bbox will only encompass your figure)
bbox = matplotlib.transforms.Bbox(((0, 0), inches))
# now you can save only the part of the figure with data
fig.savefig(savename, bbox_inches=bbox, **kwargs)

Terima kasih atas jawaban luar biasa dari semua orang ... Saya memiliki masalah yang sama persis dengan keinginan untuk merencanakan hanya gambar tanpa bantalan tambahan / ruang dll, jadi sangat senang menemukan ide semua orang di sini.

Terlepas dari gambar tanpa padding, saya juga ingin dapat dengan mudah menambahkan anotasi dll, lebih dari sekadar plot gambar sederhana.

Jadi apa yang akhirnya saya lakukan adalah menggabungkan jawaban David dengan csnemes ' untuk membuat pembungkus sederhana pada saat pembuatan gambar. Saat Anda menggunakannya, Anda tidak perlu melakukan perubahan nanti dengan imsave () atau yang lainnya:

def get_img_figure(image, dpi):
    """
    Create a matplotlib (figure,axes) for an image (numpy array) setup so that
        a) axes will span the entire figure (when saved no whitespace)
        b) when saved the figure will have the same x/y resolution as the array, 
           with the dpi value you pass in.

    Arguments:
        image -- numpy 2d array
        dpi -- dpi value that the figure should use

    Returns: (figure, ax) tuple from plt.subplots
    """

    # get required figure size in inches (reversed row/column order)
    inches = image.shape[1]/dpi, image.shape[0]/dpi

    # make figure with that size and a single axes
    fig, ax = plt.subplots(figsize=inches, dpi=dpi)

    # move axes to span entire figure area
    fig.subplots_adjust(left=0, right=1, top=1, bottom=0, wspace=0, hspace=0)

    return fig, ax