beautifyPlot: A simple matplotlib wrapper to avoid writing ‘plt.’ statements on many lines
I made a wrapper to avoid writing ‘plt.’ statements on many lines.
This method also lets me save the arguments passed to matplotlib in a file for later use.
GitHub repo: https://github.com/kvgc/beautifyPlot
Example:
matplotlib code:
import numpy as np
import matplotlib.pyplot as plt
x = np.arange(0,100,1)
y = x**3
plt.plot(x,y,label='y=f(x)')
plt.title("test plot")
plt.xlim([0,10])
plt.ylim([0,1000])
plt.legend(fontsize=15)
plt.tight_layout()
Using beautifyPlot module:
from beautifyPlot.bplt import beautifyPlot import numpy as np x = np.arange(0,100,1) y = x**3 plt.plot(x,y,label='y=f(x)') plotArgs = { 'title' : ['test plot'], ## Use list to pass a set of arguments to the function 'xlim' : [0,10], 'ylim' : [0,1000], 'legend' : {'fontsize':15, 'loc':'best'}, ## Use dictionary to specify arguments 'tight_layout' : [] ## Leave list empty if there is nothing to pass } beautifyPlot(plotArgs) ## Next steps: Save plotArgs as JSON or use the same plot arguments in a different subplot/plot.
Comments
Post a Comment