Matplotlib rakamlarımı kurtarmak için nispeten basit (ancak biraz alışılmadık) bir yol buldum. Şöyle çalışır:
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()))
şu şekilde save_plot
tanımlanmış işlev ile (mantığı anlamak için basit sürüm):
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'
veya save_plot
bunun gibi tanımlayıcı işlev (daha hafif şekil dosyaları oluşturmak için zip sıkıştırması kullanan daha iyi sürüm):
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'
Bu, libscript
çoğunlukla modüllere inspect
ve ast
. İlgi duyulursa bunu Github'da paylaşmayı deneyebilirim (önce biraz temizlik ve Github'a başlamam gerekir).
Bu save_plot
işlev ve libscript
modülün arkasındaki fikir , figürü oluşturan (modülü kullanarak inspect
) python talimatlarını almak, bunları analiz etmek (modülü kullanarak ast
), bağlı olduğu tüm değişkenleri, işlevleri ve modülleri almak, bunları yürütme bağlamından çıkarmak ve bunları serileştirmektir. python talimatları olarak (değişkenler için kod t=[0.0,2.0,0.01]
... gibi olacaktır ve modüller için kod import matplotlib.pyplot as plt
... gibi olacaktır ) şekil talimatlarının başına eklenmiştir. Ortaya çıkan python talimatları, yürütülmesi orijinal matplotlib figürünü yeniden oluşturacak bir python betiği olarak kaydedilir.
Tahmin edebileceğiniz gibi, bu matplotlib figürlerinin çoğu için (hepsi değilse de) işe yarar.