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

Popular posts from this blog

Python decorator: Example 1

Communicating and downloading data from the James Webb Space Telescope