[python] 대화 형 Matplotlib 그림 저장

나는 matplotlib 수치를 저장하는 비교적 간단한 방법 (그러나 약간 비 전통적인 방법)을 알아 냈습니다. 다음과 같이 작동합니다.

import libscript

import matplotlib.pyplot as plt
import numpy as np

t = np.arange(0.0, 2.0, 0.01)
s = 1 + np.sin(2*np.pi*t)

#<plot>
plt.plot(t, s)
plt.xlabel('time (s)')
plt.ylabel('voltage (mV)')
plt.title('About as simple as it gets, folks')
plt.grid(True)
plt.show()
#</plot>

save_plot(fileName='plot_01.py',obj=sys.argv[0],sel='plot',ctx=libscript.get_ctx(ctx_global=globals(),ctx_local=locals()))

save_plot다음과 같이 정의 된 함수 (논리를 이해하는 간단한 버전) :

def save_plot(fileName='',obj=None,sel='',ctx={}):
    """
    Save of matplolib plot to a stand alone python script containing all the data and configuration instructions to regenerate the interactive matplotlib figure.

    Parameters
    ----------
    fileName : [string] Path of the python script file to be created.
    obj : [object] Function or python object containing the lines of code to create and configure the plot to be saved.
    sel : [string] Name of the tag enclosing the lines of code to create and configure the plot to be saved.
    ctx : [dict] Dictionary containing the execution context. Values for variables not defined in the lines of code for the plot will be fetched from the context.

    Returns
    -------
    Return ``'done'`` once the plot has been saved to a python script file. This file contains all the input data and configuration to re-create the original interactive matplotlib figure.
    """
    import os
    import libscript

    N_indent=4

    src=libscript.get_src(obj=obj,sel=sel)
    src=libscript.prepend_ctx(src=src,ctx=ctx,debug=False)
    src='\n'.join([' '*N_indent+line for line in src.split('\n')])

    if(os.path.isfile(fileName)): os.remove(fileName)
    with open(fileName,'w') as f:
        f.write('import sys\n')
        f.write('sys.dont_write_bytecode=True\n')
        f.write('def main():\n')
        f.write(src+'\n')

        f.write('if(__name__=="__main__"):\n')
        f.write(' '*N_indent+'main()\n')

return 'done'

또는 다음 save_plot과 같은 기능을 정의 하십시오 (더 가벼운 그림 파일을 생성하기 위해 zip 압축을 사용하는 더 나은 버전) :

def save_plot(fileName='',obj=None,sel='',ctx={}):

    import os
    import json
    import zlib
    import base64
    import libscript

    N_indent=4
    level=9#0 to 9, default: 6
    src=libscript.get_src(obj=obj,sel=sel)
    obj=libscript.load_obj(src=src,ctx=ctx,debug=False)
    bin=base64.b64encode(zlib.compress(json.dumps(obj),level))

    if(os.path.isfile(fileName)): os.remove(fileName)
    with open(fileName,'w') as f:
        f.write('import sys\n')
        f.write('sys.dont_write_bytecode=True\n')
        f.write('def main():\n')
        f.write(' '*N_indent+'import base64\n')
        f.write(' '*N_indent+'import zlib\n')
        f.write(' '*N_indent+'import json\n')
        f.write(' '*N_indent+'import libscript\n')
        f.write(' '*N_indent+'bin="'+str(bin)+'"\n')
        f.write(' '*N_indent+'obj=json.loads(zlib.decompress(base64.b64decode(bin)))\n')
        f.write(' '*N_indent+'libscript.exec_obj(obj=obj,tempfile=False)\n')

        f.write('if(__name__=="__main__"):\n')
        f.write(' '*N_indent+'main()\n')

return 'done'

이 차종은 모듈을 사용하는 libscript대부분의 모듈에 의존하는, 내 자신의 inspect과 ast. 관심이 있으면 Github에서 공유해 볼 수 있습니다 (먼저 정리가 필요하고 Github를 시작하려면 제가 필요합니다).

이 save_plot함수와 libscript모듈의 이면에있는 아이디어 는 Figure를 생성 (module 사용 inspect)하고 분석 (module 사용 ast)하여 모든 변수, 함수 및 모듈 가져 오기를 추출하고 실행 컨텍스트에서이를 추출하고 직렬화 하는 Python 명령어를 가져 오는 것입니다. 파이썬 명령어 (변수에 대한 코드는 t=[0.0,2.0,0.01]…와 같고 모듈에 대한 코드는 …와 같음 import matplotlib.pyplot as plt)로 그림 명령어 앞에 추가됩니다. 결과 python 명령어는 실행이 원래 matplotlib 그림을 다시 빌드하는 python 스크립트로 저장됩니다.

상상할 수 있듯이 이것은 대부분의 (전부는 아니지만) matplotlib 그림에서 잘 작동합니다.

답변