Matplotlib grafikleri: eksen, efsaneler ve beyaz boşlukların kaldırılması

Python ve Matplotlib'de yeniyim, eksenlere, etiketlere, başlıklara veya genellikle matplotlib tarafından otomatik olarak eklenen herhangi bir şey kullanmadan, bir resme renk haritasını uygulamak ve elde edilen görüntüyü yazmak istiyorum. İşte yaptığım şey:

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)

Şeklin eksenini başarıyla kaldırır, ancak kaydedilen şekil beyaz bir dolgu ve gerçek görüntünün etrafında bir çerçeve sunar. Bunları nasıl çıkarabilirim (en azından beyaz dolgu)? Teşekkürler

Komutun axis('off'), sorunlardan birini, her ekseni ve kenarlığı ayrı ayrı değiştirmekten daha özlü bir şekilde ele aldığını düşünüyorum . Yine de sınır çevresindeki beyaz boşluğu bırakıyor. Ekleme bbox_inches='tight'için savefigbeyaz boşluk sol çok daha küçük, ama yine de mevcut olduğunu aşağıda komutla neredeyse orada alır, sen örnekte görebilirsiniz.

bbox_inches=0Dize yerine matplotlib'in daha yeni sürümlerinin gerekli olabileceğini unutmayın 'tight'(@episodeyang ve @kadrach aracılığıyla)

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

Bu numarayı matehat'tan öğrendim , burada :

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

verim

Olası en basit çözüm:

Ben sadece soruda açıklanan yöntemi ve Hooked tarafından cevap yöntemini birleştirdim.

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)

Bu koddan sonra boşluk ve çerçeve yoktur.

Henüz kimseden bahsetmedi imsave, bu da onu bir astar haline getiriyor:

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

Görüntüyü olduğu gibi doğrudan saklar, yani herhangi bir eksen veya kenarlık / dolgu eklemez.

Ayrıca, şeklin kapsamını bbox_inchesbağımsız değişkene de belirleyebilirsiniz . Bu, figürün etrafındaki beyaz dolgudan kurtulacaktır.

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)

Bu, tüm dolguları ve kenarlıkları kaldırmalıdır:

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)

Her şeyin belgelendiğini buldum ...

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

Kodum…. "bcK" 512x512 görüntü

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

Yükseltilmiş cevap artık çalışmıyor. Çalışması için [0, 0, 1, 1] olarak ayarlanmış bir ekseni manuel olarak eklemeniz veya şekil altındaki yamayı kaldırmanız gerekir.

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

Alternatif olarak, yamayı kaldırabilirsiniz. Dolguları kaldırmak için bir alt grafik eklemenize gerek yoktur. Bu, Vlady'nin aşağıdaki cevabından basitleştirilmiştir

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

Bu, 3.0.32019/06/19 sürümü ile test edilmiştir . Görüntü aşağıdaki feryat:

Yapılacak daha basit bir şey kullanmaktır pyplot.imsave. Ayrıntılar için aşağıdaki luator yanıtına bakın

Ubuntu'nun cevabını beğendim , ancak kare olmayan görüntüler için kutunun dışında nasıl ayarlanacağını açıkça göstermiyordum, bu yüzden kolay kopyala yapıştır için değiştirdim:

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 = boyut korunuyorsa, seçtiğiniz dpi ne olursa olsun görüntüleri kenarlıksız kaydetmek kolaydır.

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

Ancak görüntüleme ürkütücü. Küçük dpi seçerseniz, görüntü boyutunuz ekranınızdan daha büyük olabilir ve ekran sırasında kenarlık elde edebilirsiniz. Bununla birlikte, bu tasarrufu etkilemez.

İçin böylece

save_image_fix_dpi(data, dpi=20)

Ekran sınırlanır (ancak kaydetme çalışır):

İlk olarak, belirli görüntü formatları (yani TIFF ) için renk haritasını başlığa kaydedebilirsiniz ve çoğu izleyici verilerinizi renk haritasıyla gösterecektir.

matplotlibGörüntülere ek açıklamalar veya başka veriler eklemek için yararlı olabilecek gerçek bir resmi kaydetmek için aşağıdaki çözümü kullandım:

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)

Herkesten harika cevaplar için teşekkürler ... Ben sadece ekstra bir dolgu / boşluk vb ile bir görüntü çizmek isteyen ile aynı sorunu vardı, bu yüzden burada herkesin fikirlerini bulmak için çok mutlu oldu.

Hiçbir dolgu ile görüntü dışında, sadece basit bir görüntü arsa ötesinde kolayca ek açıklamalar vb eklemek istedim.

Sonuçta David'in cevabını csnemes ile birleştirerek figür yaratma zamanında basit bir sarıcı yapmaktı. Bunu kullandığınızda, daha sonra imsave () veya başka bir şeyle değişiklik yapmanız gerekmez:

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