Lô Matplotlib: loại bỏ trục, truyền thuyết và khoảng trắng

Tôi mới sử dụng Python và Matplotlib, tôi chỉ muốn áp dụng colormap cho một hình ảnh và viết hình ảnh kết quả, mà không sử dụng trục, nhãn, tiêu đề hoặc bất cứ thứ gì thường được tự động thêm bởi matplotlib. Đây là những gì tôi đã làm:

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)

Nó loại bỏ thành công trục của hình, nhưng hình được lưu thể hiện phần đệm trắng và khung xung quanh ảnh thực. Làm thế nào tôi có thể loại bỏ chúng (ít nhất là phần đệm trắng)? Cảm ơn

Tôi nghĩ rằng lệnh axis('off')này quan tâm đến một trong những vấn đề ngắn gọn hơn là thay đổi từng trục và đường viền riêng biệt. Nó vẫn để lại khoảng trắng xung quanh biên giới. Thêm bbox_inches='tight'vào savefiglệnh gần như đưa bạn đến đó, bạn có thể thấy trong ví dụ bên dưới rằng khoảng trắng còn lại nhỏ hơn nhiều, nhưng vẫn hiện diện.

Lưu ý rằng các phiên bản mới hơn của matplotlib có thể yêu cầu bbox_inches=0thay vì chuỗi 'tight'(thông qua @epiTyang và @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')

Tôi đã học được mẹo này từ matehat, ở đây :

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

sản lượng

Giải pháp đơn giản nhất có thể:

Tôi chỉ đơn giản là kết hợp phương pháp được mô tả trong câu hỏi và phương pháp từ câu trả lời của 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)

Sau mã này không có khoảng trắng và không có khung.

Không ai đề cập đến imsave, điều này làm cho điều này một lót:

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

Nó trực tiếp lưu trữ hình ảnh như hiện tại, tức là không thêm bất kỳ trục hoặc đường viền / phần đệm nào.

Bạn cũng có thể chỉ định phạm vi của hình cho bbox_inchesđối số. Điều này sẽ thoát khỏi phần đệm trắng xung quanh hình.

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)

Điều này sẽ loại bỏ tất cả các phần đệm và đường viền:

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)

Tôi thấy rằng tất cả đều được ghi lại ...

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

Mã của tôi…. "bcK" là hình ảnh 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()

Câu trả lời nâng cao không hoạt động nữa. Để làm cho nó hoạt động, bạn cần thêm thủ công một trục được đặt thành [0, 0, 1, 1] hoặc xóa bản vá trong hình.

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

Ngoài ra, bạn chỉ có thể loại bỏ các bản vá. Bạn không cần phải thêm một subplot để loại bỏ phần đệm. Điều này được đơn giản hóa từ câu trả lời của Vladian dưới đây

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

Điều này đã được thử nghiệm với phiên bản 3.0.3vào ngày 2019/06/19. Hình ảnh xem dưới đây:

Một điều đơn giản hơn nhiều để làm là sử dụng pyplot.imsave. Để biết chi tiết, xem dưới đây câu trả lời của luator

Tôi thích câu trả lời của Ubuntu , nhưng nó không hiển thị rõ ràng cách đặt kích thước cho hình ảnh không vuông, vì vậy tôi đã sửa đổi nó để dễ sao chép:

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

Lưu hình ảnh không có viền là dễ dàng bất cứ dpi nào bạn chọn nếu pixel_size / dpi = size được giữ.

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

Tuy nhiên hiển thị là ma quái. Nếu bạn chọn dpi nhỏ, kích thước hình ảnh của bạn có thể lớn hơn màn hình của bạn và bạn có đường viền trong khi hiển thị. Tuy nhiên, điều này không ảnh hưởng đến tiết kiệm.

Vì vậy đối với

save_image_fix_dpi(data, dpi=20)

Màn hình trở nên viền (nhưng lưu công việc):

Đầu tiên, đối với các định dạng hình ảnh nhất định (ví dụ TIFF ), bạn thực sự có thể lưu bản đồ màu trong tiêu đề và hầu hết người xem sẽ hiển thị dữ liệu của bạn bằng bản đồ màu.

Để lưu matplotlibhình ảnh thực tế , có thể hữu ích để thêm chú thích hoặc dữ liệu khác vào hình ảnh, tôi đã sử dụng giải pháp sau:

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)

Cảm ơn câu trả lời tuyệt vời từ mọi người ... Tôi có cùng một vấn đề với việc chỉ muốn vẽ một hình ảnh không có phần đệm / không gian thêm, v.v., vì vậy tôi đã rất vui khi tìm thấy ý tưởng của mọi người ở đây.

Ngoài hình ảnh không có phần đệm, tôi cũng muốn có thể dễ dàng thêm chú thích, v.v., ngoài một cốt truyện hình ảnh đơn giản.

Vì vậy, những gì tôi đã làm là kết hợp câu trả lời của David với csneme ' để tạo ra một trình bao bọc đơn giản tại thời điểm tạo hình. Khi bạn sử dụng điều đó, bạn không cần bất kỳ thay đổi nào sau đó với imsave () hoặc bất cứ điều gì khác:

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