Видалення білого простору навколо збереженого зображення в matplotlib

Мені потрібно сфотографувати і зберегти його після деякого процесу. Фігура виглядає добре, коли я її відображаю, але після збереження фігури я отримав пробіл навколо збереженого зображення. Я спробував 'tight'варіант для savefigметоду, теж не працював. Код:

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

Я намагаюся намалювати базовий графік, використовуючи NetworkX на фігурі, і зберегти його. Я зрозумів, що без графіку він працює, але, додавши графік, я отримую пробіл навколо збереженого зображення;

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

Я не можу стверджувати, що точно знаю, чому або як працює моє «рішення», але це те, що мені довелося зробити, коли я хотів скласти контур пари секцій аерофолів - без білих полів - у файл PDF. (Зверніть увагу, що я використовував matplotlib всередині ноутбука IPython із прапором -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)

Я намагався деактивувати різні частини цього, але це завжди десь призводить до білого поля. Ви можете навіть змінити це, щоб жирові лінії не були витіснені від меж фігури від нестачі запасів.

Ви можете видалити порожній простір відступів шляхом установки bbox_inches="tight"в savefig:

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

Вам доведеться поставити аргумент bbox_inchesяк рядок, можливо, саме тому він не працював раніше для вас.

Можливі дублікати:

Сюжети Matplotlib: видалення осі, легенди та пробіли

Як встановити поля для фігури matplotlib?

Зменшити лівий і правий поля в графіку matplotlib

Після спроби вищезазначених відповідей без успіху (і ряду інших ставок) те, що нарешті працювало для мене, було просто

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

Що важливо, це не включає аргументи bbox або padding.

Я знайшов щось від Арвінда Перейра ( http://robotics.usc.edu/~ampereir/wordpress/?p=626 ) і, здається, працює для мене:

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

Наступна функція включає в себе відповідь johannes-s вище. Я перевірив його plt.figureі plt.subplots()з декількома осями, і він прекрасно працює.

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

Я виявив, що наступні коди прекрасно працюють для цієї роботи.

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)

Я дотримувався цієї послідовності, і це працювало як шарм.

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)

Для всіх, хто хоче працювати в пікселях, а не в дюймах, це буде працювати.

Плюс звичайне вам також знадобиться

from matplotlib.transforms import Bbox

Тоді ви можете використовувати наступне:

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)

Набагато простішим підходом я став plt.imsave:

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

Ви можете спробувати це. Це вирішило моє питання.

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

Якщо ви хочете показати, що потрібно зберегти, я рекомендую використовувати plt.tight_layoutтрансформацію, яка є фактично більш кращою, оскільки вона не робить зайвого обрізання при використанніplt.savefig

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

Це працює для мене, зберігаючи numpy масив, побудований з імшоу для файлу

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