Xóa khoảng trắng xung quanh ảnh đã lưu trong matplotlib

Tôi cần phải chụp ảnh và lưu nó sau một quá trình. Con số trông ổn khi tôi hiển thị nó, nhưng sau khi lưu hình, tôi có một khoảng trắng xung quanh ảnh đã lưu. Tôi đã thử 'tight'tùy chọn cho savefigphương thức, cũng không hoạt động. Mật mã:

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

  fig = plt.figure(1)
  img = mpimg.imread(path)
  plt.imshow(img)
  ax=fig.add_subplot(1,1,1)

  extent = ax.get_window_extent().transformed(fig.dpi_scale_trans.inverted())
  plt.savefig('1.png', bbox_inches=extent)

  plt.axis('off') 
  plt.show()

Tôi đang cố gắng vẽ một biểu đồ cơ bản bằng cách sử dụng NetworkX trên một hình và lưu nó. Tôi nhận ra rằng không có biểu đồ, nó hoạt động, nhưng khi thêm biểu đồ tôi sẽ có khoảng trắng xung quanh ảnh đã lưu;

import matplotlib.image as mpimg
import matplotlib.pyplot as plt
import networkx as nx

G = nx.Graph()
G.add_node(1)
G.add_node(2)
G.add_node(3)
G.add_edge(1,3)
G.add_edge(1,2)
pos = {1:[100,120], 2:[200,300], 3:[50,75]}

fig = plt.figure(1)
img = mpimg.imread("C:\\images\\1.jpg")
plt.imshow(img)
ax=fig.add_subplot(1,1,1)

nx.draw(G, pos=pos)

extent = ax.get_window_extent().transformed(fig.dpi_scale_trans.inverted())
plt.savefig('1.png', bbox_inches = extent)

plt.axis('off') 
plt.show()

Tôi không thể khẳng định tôi biết chính xác lý do tại sao hoặc cách thức giải pháp trên máy tính của tôi, nhưng đây là điều tôi phải làm khi tôi muốn phác thảo phác thảo của một vài phần của aerofoil - không có lề trắng - vào tệp PDF. (Lưu ý rằng tôi đã sử dụng matplotlib trong sổ ghi chép IPython, với cờ -pylab.)

plt.gca().set_axis_off()
plt.subplots_adjust(top = 1, bottom = 0, right = 1, left = 0, 
            hspace = 0, wspace = 0)
plt.margins(0,0)
plt.gca().xaxis.set_major_locator(plt.NullLocator())
plt.gca().yaxis.set_major_locator(plt.NullLocator())
plt.savefig("filename.pdf", bbox_inches = 'tight',
    pad_inches = 0)

Tôi đã cố gắng hủy kích hoạt các phần khác nhau của điều này, nhưng điều này luôn dẫn đến một lề trắng ở đâu đó. Bạn thậm chí có thể sửa đổi điều này để giữ cho các đường mỡ gần giới hạn của hình không bị cạo do thiếu lề.

Bạn có thể xóa phần đệm trắng bằng cách đặt bbox_inches="tight"vào savefig:

plt.savefig("test.png",bbox_inches='tight')

Bạn sẽ phải đặt đối số thành bbox_inchesmột chuỗi, có lẽ đây là lý do tại sao nó không hoạt động sớm hơn đối với bạn.

Các bản sao có thể có:

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

Làm thế nào để thiết lập lề cho một con số matplotlib?

Giảm lề trái và phải trong lô matplotlib

Sau khi thử các câu trả lời trên mà không thành công (và một loạt các bài đăng ngăn xếp khác), cuối cùng tôi cũng làm được

plt.gca().set_axis_off()
plt.subplots_adjust(top = 1, bottom = 0, right = 1, left = 0, 
            hspace = 0, wspace = 0)
plt.margins(0,0)
plt.savefig("myfig.pdf")

Điều quan trọng này không bao gồm các đối số bbox hoặc padding.

Tôi đã tìm thấy thứ gì đó từ Arvind Pereira ( http://robotics.usc.edu/~ampereir/wordpress/?p=626 ) và dường như hoạt động với tôi:

plt.savefig(filename, transparent = True, bbox_inches = 'tight', pad_inches = 0)

Các chức năng sau đây kết hợp câu trả lời johannes-s ở trên. Tôi đã thử nghiệm nó với plt.figurevà plt.subplots()với nhiều trục, và nó hoạt động độc đáo.

def save(filepath, fig=None):
    '''Save the current image with no whitespace
    Example filepath: "myfig.png" or r"C:\myfig.pdf" 
    '''
    import matplotlib.pyplot as plt
    if not fig:
        fig = plt.gcf()

    plt.subplots_adjust(0,0,1,1,0,0)
    for ax in fig.axes:
        ax.axis('off')
        ax.margins(0,0)
        ax.xaxis.set_major_locator(plt.NullLocator())
        ax.yaxis.set_major_locator(plt.NullLocator())
    fig.savefig(filepath, pad_inches = 0, bbox_inches='tight')

Tôi tìm thấy các mã sau đây hoạt động hoàn hảo cho công việc.

fig = plt.figure(figsize=[6,6])
ax = fig.add_subplot(111)
ax.imshow(data)
ax.axes.get_xaxis().set_visible(False)
ax.axes.get_yaxis().set_visible(False)
ax.set_frame_on(False)
plt.savefig('data.png', dpi=400, bbox_inches='tight',pad_inches=0)

tôi theo trình tự này và nó hoạt động như một lá bùa.

plt.axis("off")
fig=plt.imshow(image array,interpolation='nearest')
fig.axes.get_xaxis().set_visible(False)
fig.axes.get_yaxis().set_visible(False)
plt.savefig('destination_path.pdf',
    bbox_inches='tight', pad_inches=0, format='pdf', dpi=1200)

Đối với bất cứ ai muốn làm việc bằng pixel thay vì inch thì điều này sẽ hoạt động.

Cộng với thông thường bạn cũng sẽ cần

from matplotlib.transforms import Bbox

Sau đó, bạn có thể sử dụng như sau:

my_dpi = 100 # Good default - doesn't really matter

# Size of output in pixels
h = 224
w = 224

fig, ax = plt.subplots(1, figsize=(w/my_dpi, h/my_dpi), dpi=my_dpi)

ax.set_position([0, 0, 1, 1]) # Critical!

# Do some stuff
ax.imshow(img)
ax.imshow(heatmap) # 4-channel RGBA
ax.plot([50, 100, 150], [50, 100, 150], color="red")

ax.axis("off")

fig.savefig("saved_img.png",
            bbox_inches=Bbox([[0, 0], [w/my_dpi, h/my_dpi]]),
            dpi=my_dpi)

Một cách tiếp cận đơn giản hơn nhiều tôi tìm thấy là sử dụng plt.imsave:

    import matplotlib.pyplot as plt
    arr = plt.imread(path)
    plt.imsave('test.png', arr)

Bạn có thể thử điều này. Nó giải quyết vấn đề của tôi.

import matplotlib.image as mpimg
img = mpimg.imread("src.png")
mpimg.imsave("out.png", img, cmap=cmap)

Nếu bạn muốn hiển thị những gì sẽ được lưu, tôi khuyên bạn nên sử dụng plt.tight_layoutchuyển đổi, điều này thực sự thích hợp hơn vì nó không thực hiện cắt xén không cần thiết khi sử dụngplt.savefig

import matplotlib as plt    
plt.plot([1,2,3], [1,2,3])
plt.tight_layout(pad=0)
plt.savefig('plot.png')

Điều này làm việc cho tôi để lưu một mảng numpy được vẽ với imshow để tập tin

import matplotlib.pyplot as plt

fig = plt.figure(figsize=(10,10))
plt.imshow(img) # your image here
plt.axis("off")
plt.subplots_adjust(top = 1, bottom = 0, right = 1, left = 0, 
        hspace = 0, wspace = 0)
plt.savefig("example2.png", box_inches='tight', dpi=100)
plt.show()