Matplotlib를 사용하여 플롯을 이미지 파일로 표시하지 않고 이미지 파일로 저장

나는 플롯을 즉시 생성하기 위해 빠르고 더러운 스크립트를 작성하고 있습니다. Matplotlib 설명서의 아래 코드를 시작점으로 사용하고 있습니다.

from pylab import figure, axes, pie, title, show

# Make a square figure and axes
figure(1, figsize=(6, 6))
ax = axes([0.1, 0.1, 0.8, 0.8])

labels = 'Frogs', 'Hogs', 'Dogs', 'Logs'
fracs = [15, 30, 45, 10]

explode = (0, 0.05, 0, 0)
pie(fracs, explode=explode, labels=labels, autopct='%1.1f%%', shadow=True)
title('Raining Hogs and Dogs', bbox={'facecolor': '0.8', 'pad': 5})

show()  # Actually, don't show, just save to foo.png

GUI에 플롯을 표시하고 싶지 않고 대신 플롯을 파일 (foo.png)로 저장하여 배치 스크립트에서 사용할 수 있습니다. 어떻게합니까?

질문에 대한 답변이 있지만 matplotlib.pyplot.savefig 사용할 때 유용한 팁을 추가하고 싶습니다 . 파일 형식은 확장자로 지정할 수 있습니다.

from matplotlib import pyplot as plt

plt.savefig('foo.png')
plt.savefig('foo.pdf')

래스터 화 된 또는 벡터화 된 출력을 각각 제공 할 수 있습니다. 또한 pylab이미지 주위에 관대하고 종종 바람직하지 않은 공백이 남는 것을 알 수 있습니다. 다음으로 제거하십시오.

savefig('foo.png', bbox_inches='tight')

다른 사람이 말했다, 한 plt.savefig()또는 fig1.savefig()참으로 이미지를 저장하는 방법입니다.

그러나 어떤 경우 에는 항상 그림이 표시 됩니다. (예 plt.ion(): Spyder with 대화 형 모드 = 켜기) 나는 거대한 루프에서 그림 창을 강제로 닫아서이 문제를 해결 하므로 plt.close(figure_object)( 문서 참조 ) 루프 중에 백만 개의 열린 인물이 없습니다.

import matplotlib.pyplot as plt
fig, ax = plt.subplots( nrows=1, ncols=1 )  # create figure & 1 axis
ax.plot([0,1,2], [10,20,3])
fig.savefig('path/to/save/image/to.png')   # save the figure to file
plt.close(fig)    # close the figure window

필요한 경우 나중에 그림을 다시 열 수 있어야합니다 fig.show()(자신을 테스트하지 않았습니다).

해결책은 다음과 같습니다.

pylab.savefig('foo.png')

이 문제를 해결하기 위해 MatPlotLib 설명서에서이 링크를 찾았습니다. http://matplotlib.org/faq/howto_faq.html#generate-images-without-having-a-window-appear

그들은 그림이 팝업되는 것을 막는 가장 쉬운 방법은 다음과 같이 비대화 형 백엔드 (예 : Agg)를 사용하는 것입니다 matplotib.use(<backend>).

import matplotlib
matplotlib.use('Agg')
import matplotlib.pyplot as plt
plt.plot([1,2,3])
plt.savefig('myfig')

plt.close( fig )루프를 사용하는 동안 특정 그림을 숨길 수있는 옵션이 있지만 여전히 루프 후 데이터 처리를 위해 그림을 표시하기 때문에 개인적으로을 선호합니다 . 비록 비대화 형 백엔드를 선택하는 것보다 느릴 것입니다. 누군가 그것을 테스트하면 흥미로울 것입니다.

업데이트 : Spyder의 경우 일반적으로 이러한 방식으로 백엔드를 설정할 수 없습니다 (Spyder는 일반적으로 matplotlib를 일찍로드하여을 사용하지 못하기 때문에 matplotlib.use()).

대신 Spyder prefs에서 plt.switch_backend('Agg')" enable support "를 사용 하거나 끄고 matplotlib.use('Agg')직접 명령을 실행하십시오.

이 두 힌트에서 : 하나 , 둘

"현재"그림의 개념이 마음에 들지 않으면 다음을 수행하십시오.

import matplotlib.image as mpimg

img = mpimg.imread("src.png")
mpimg.imsave("out.png", img)

다른 답변은 맞습니다. 그러나 때때로 그림 개체 를 열고 싶다는 것을 알게 되었습니다. 예를 들어, 레이블 크기를 변경하거나 그리드를 추가하거나 다른 처리를 원할 수 있습니다. 완벽한 세상에서 나는 줄거리를 생성하는 코드를 다시 실행하고 설정을 조정합니다. 아아, 세상은 완벽하지 않다. 따라서 PDF 또는 PNG로 저장하는 것 외에도 다음을 추가합니다.

with open('some_file.pkl', "wb") as fp:
    pickle.dump(fig, fp, protocol=4)

이처럼 나중에 그림 개체를로드하고 원하는대로 설정을 조작 할 수 있습니다.

또한 스택의 locals()각 함수 / 방법에 대한 소스 코드와 사전을 사용하여 스택을 작성하여 나중에 그림을 생성 한 대상을 정확하게 알 수 있습니다.

주의 : 때때로이 방법은 큰 파일을 생성하므로주의하십시오.

import datetime
import numpy as np
from matplotlib.backends.backend_pdf import PdfPages
import matplotlib.pyplot as plt

# Create the PdfPages object to which we will save the pages:
# The with statement makes sure that the PdfPages object is closed properly at
# the end of the block, even if an Exception occurs.
with PdfPages('multipage_pdf.pdf') as pdf:
    plt.figure(figsize=(3, 3))
    plt.plot(range(7), [3, 1, 4, 1, 5, 9, 2], 'r-o')
    plt.title('Page One')
    pdf.savefig()  # saves the current figure into a pdf page
    plt.close()

    plt.rc('text', usetex=True)
    plt.figure(figsize=(8, 6))
    x = np.arange(0, 5, 0.1)
    plt.plot(x, np.sin(x), 'b-')
    plt.title('Page Two')
    pdf.savefig()
    plt.close()

    plt.rc('text', usetex=False)
    fig = plt.figure(figsize=(4, 5))
    plt.plot(x, x*x, 'ko')
    plt.title('Page Three')
    pdf.savefig(fig)  # or you can pass a Figure object to pdf.savefig
    plt.close()

    # We can also set the file's metadata via the PdfPages object:
    d = pdf.infodict()
    d['Title'] = 'Multipage PDF Example'
    d['Author'] = u'Jouni K. Sepp\xe4nen'
    d['Subject'] = 'How to create a multipage pdf file and set its metadata'
    d['Keywords'] = 'PdfPages multipage keywords author title subject'
    d['CreationDate'] = datetime.datetime(2009, 11, 13)
    d['ModDate'] = datetime.datetime.today()

plot () 및 기타 함수를 사용하여 원하는 내용을 만든 후 다음과 같은 절을 사용하여 화면 또는 파일에 플로팅 중에서 선택할 수 있습니다.

import matplotlib.pyplot as plt

fig = plt.figure(figsize=(4, 5))       # size in inches
# use plot(), etc. to create your plot.

# Pick one of the following lines to uncomment
# save_file = None
# save_file = os.path.join(your_directory, your_file_name)  

if save_file:
    plt.savefig(save_file)
    plt.close(fig)
else:
    plt.show()

나는 다음을 사용했다.

import matplotlib.pyplot as plt

p1 = plt.plot(dates, temp, 'r-', label="Temperature (celsius)")  
p2 = plt.plot(dates, psal, 'b-', label="Salinity (psu)")  
plt.legend(loc='upper center', numpoints=1, bbox_to_anchor=(0.5, -0.05),        ncol=2, fancybox=True, shadow=True)

plt.savefig('data.png')  
plt.show()  
f.close()
plt.close()

그림을 저장 한 후 plt.show를 사용하는 것이 매우 중요하다는 것을 알았습니다. 그렇지 않으면 작동하지 않습니다. png로 내 보낸 도형

당신은 할 수 있습니다 :

plt.show(hold=False)
plt.savefig('name.pdf')

GUI 플롯을 닫기 전에 savefig가 완료되도록해야합니다. 이런 식으로 이미지를 미리 볼 수 있습니다.

양자 택일로, 당신은 그것을 볼 수 있습니다 plt.show() 그리고 가까운 GUI 및 스크립트를 다시 실행하지만, 이번에 교체 plt.show()와 함께 plt.savefig().

또는 사용할 수 있습니다

fig, ax = plt.figure(nrows=1, ncols=1)
plt.plot(...)
plt.show()
fig.savefig('out.pdf')

나와 같이 Spyder IDE를 사용하는 경우 다음을 사용하여 대화식 모드를 비활성화해야합니다.

plt.ioff()

(이 명령은 과학적 시작과 함께 자동으로 시작됩니다)

다시 활성화하려면 다음을 사용하십시오.

plt.ion()

해결책 :

import pandas as pd
import numpy as np
import matplotlib.pyplot as plt
import matplotlib
matplotlib.style.use('ggplot')
ts = pd.Series(np.random.randn(1000), index=pd.date_range('1/1/2000', periods=1000))
ts = ts.cumsum()
plt.figure()
ts.plot()
plt.savefig("foo.png", bbox_inches='tight')

이미지를 저장하고 이미지를 저장하려면 다음을 사용하십시오.

%matplotlib inline

후 import matplotlib

질문 Matplotlib (pyplot) 에 따르면 savefig는 빈 이미지를 출력합니다 .

한 가지주의해야 할 점은 사용 plt.show하고 이후 plt.savefig에 사용해야 할 경우 빈 이미지를 제공하는 것입니다.

자세한 예 :

import numpy as np
import matplotlib.pyplot as plt


def draw_result(lst_iter, lst_loss, lst_acc, title):
    plt.plot(lst_iter, lst_loss, '-b', label='loss')
    plt.plot(lst_iter, lst_acc, '-r', label='accuracy')

    plt.xlabel("n iteration")
    plt.legend(loc='upper left')
    plt.title(title)
    plt.savefig(title+".png")  # should before plt.show method

    plt.show()


def test_draw():
    lst_iter = range(100)
    lst_loss = [0.01 * i + 0.01 * i ** 2 for i in xrange(100)]
    # lst_loss = np.random.randn(1, 100).reshape((100, ))
    lst_acc = [0.01 * i - 0.01 * i ** 2 for i in xrange(100)]
    # lst_acc = np.random.randn(1, 100).reshape((100, ))
    draw_result(lst_iter, lst_loss, lst_acc, "sgd_method")


if __name__ == '__main__':
    test_draw()

import matplotlib.pyplot as plt
plt.savefig("image.png")

Jupyter Notebook에서는 한 셀의 나머지 plt 코드와 함께을 제거 plt.show()하고 추가 해야합니다 plt.savefig(). 이미지는 여전히 노트북에 나타납니다.

감안할 때 오늘 (이 질문이 이루어진 경우에 사용할 수없는) 많은 사람들이 같은 플롯을 저장하는 매우 쉬운 방법이, 파이썬 콘솔로 Jupyter 노트북을 사용하여 .png바로 전화 matplotlib의 pylab그림을 플롯 Jupyter 노트북에서 클래스 '인라인 'jupyter 셀을 찾은 다음 해당 그림 / 이미지를 로컬 디렉토리로 드래그하십시오. %matplotlib inline첫 줄을 잊지 마십시오 !

위의 것 외에도 __file__그림과 Python 파일의 이름이 동일하도록 이름을 추가했습니다. 또한 더 나아 보이도록 몇 가지 주장을 추가했습니다.

# Saves a PNG file of the current graph to the folder and updates it every time
# (nameOfimage, dpi=(sizeOfimage),Keeps_Labels_From_Disappearing)
plt.savefig(__file__+".png",dpi=(250), bbox_inches='tight')
# Hard coded name: './test.png'

를 사용할 때 matplotlib.pyplot먼저 플롯을 저장 한 후 다음 두 줄을 사용하여 플롯을 닫아야합니다.

fig.savefig('plot.png') # save the plot, place the path you want to save the figure in quotation
plt.close(fig) # close the figure window

이전에 제안했듯이 다음 중 하나를 사용할 수 있습니다.

import matplotlib.pyplot as plt
plt.savefig("myfig.png")

표시중인 IPhython 이미지를 저장합니다. 또는 다른 음으로 (다른 각도에서 보았을 때) 열린 cv로 작업하거나 열린 cv를 가져온 경우 다음을 수행 할 수 있습니다.

수입 cv2

cv2.imwrite ( "myfig.png", 이미지)

그러나 Open CV로 작업 해야하는 경우에 대비합니다. 그렇지 않으면 plt.savefig ()로 충분합니다.

확장명 (png, jpg 등)과 원하는 해상도로 이미지를 저장할 수 있습니다. 그림을 저장하는 기능이 있습니다.

import os

def save_fig(fig_id, tight_layout=True, fig_extension="png", resolution=300):
    path = os.path.join(IMAGES_PATH, fig_id + "." + fig_extension)
    print("Saving figure", fig_id)
    if tight_layout:
        plt.tight_layout()
    plt.savefig(path, format=fig_extension, dpi=resolution)

'fig_id'는 그림을 저장하려는 이름입니다. 그것이 도움이되기를 바랍니다 :)

다음과 같이 할 수 있습니다 :

def plotAFig():
  plt.figure()
  plt.plot(x,y,'b-')
  plt.savefig("figurename.png")
  plt.close()