मुझे अपने मैटलपोटलिब के आंकड़ों को बचाने के लिए एक अपेक्षाकृत सरल तरीका (अभी तक थोड़ा अपरंपरागत) लगा। यह इस तरह काम करता है:
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
इस तरह के कार्य को परिभाषित करना (लाइटर फिगर फ़ाइलों का उत्पादन करने के लिए ज़िप संपीड़न का उपयोग करके बेहतर संस्करण):
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
। यदि ब्याज व्यक्त किया जाता है, तो मैं इसे गितुब पर साझा करने का प्रयास कर सकता हूं (इसे पहले कुछ सफाई की आवश्यकता होगी और मुझे गितूब के साथ शुरू करने की आवश्यकता होगी)।
इस save_plot
फ़ंक्शन और libscript
मॉड्यूल के पीछे का विचार अजगर के निर्देशों को प्राप्त करना है जो आंकड़ा (मॉड्यूल का उपयोग करके inspect
) बनाते हैं , उन्हें विश्लेषण करते हैं (मॉड्यूल का उपयोग करके ast
) सभी चर, फ़ंक्शन और मॉड्यूल आयात करते हैं जो इस पर निर्भर करते हैं, निष्पादन संदर्भ से इन्हें निकालें और इन्हें क्रमबद्ध करें। अजगर निर्देशों के रूप में (चर के लिए कोड की तरह होगा t=[0.0,2.0,0.01]
... और मॉड्यूल के लिए कोड की तरह होगा import matplotlib.pyplot as plt
...) आंकड़ा निर्देशों के लिए तैयार है। परिणामस्वरूप अजगर निर्देशों को एक अजगर स्क्रिप्ट के रूप में सहेजा जाता है, जिसका निष्पादन मूल matplotlib आकृति को फिर से बनाएगा।
जैसा कि आप कल्पना कर सकते हैं, यह सबसे (अगर सभी के लिए नहीं) matplotlib आंकड़ों के लिए अच्छी तरह से काम करता है।