Produce publication-level quality images on top of Matplotlib, with a simple call to a couple functions at the start and end of your script.
For similar librairies, see the References section.
pip install publib
At the beginning of the script, call:
set_style()
After each new axe is plotted, call:
fix_style()
Note that importing publib will already load the basic style.
A few more styles ('poster', 'article', etc.) can be selected with the
function set_style()
Because some matplotlib parameters cannot be changed before the lines
are plotted, they are called through the function fix_style() which:
-
changes the minor ticks
-
remove the spines
-
turn the legend draggable by default
import numpy as np
import matplotlib.pyplot as plt
import matplotlib as mpl
A default Matplotlib plot:
mpl.rcdefaults()
x = np.linspace(0,5,250)
y = np.cos(x)**2+np.random.normal(scale=0.5,size=len(x))
yav = np.cos(x)**2
plt.figure()
ax = plt.subplot()
ax.plot(x,y,'o',label='normal distribution')
ax.plot(x,yav,zorder=-1,label='average')
plt.xlabel(r'$x$')
plt.ylabel(r'$\cos^2 x$+noise')
plt.title('matplotlib')
plt.legend(loc='upper left')
plt.ylim((-1.5,3.5))
plt.show()
plt.savefig('mpl_default.png')
And now the same code with the two new lines calling the publib functions
from publib import set_style, fix_style
set_style('article') # before the first plot
x = np.linspace(0,5,250)
y = np.cos(x)**2+np.random.normal(scale=0.5,size=len(x))
yav = np.cos(x)**2
plt.figure()
ax = plt.subplot()
ax.plot(x,y,'o',label='normal distribution')
ax.plot(x,yav,zorder=-1,label='average')
plt.xlabel(r'$x$')
plt.ylabel(r'$\cos^2 x$+noise')
plt.title('article')
plt.legend(loc='upper left')
plt.ylim((-1.5,3.5))
fix_style('article') # after the axe has been created
plt.show()
plt.savefig('publib_article.png')
The OriginPro style:
set_style('origin')
...
fix_style('origin')
A combination of styles:
set_style(['poster', 'origin'])
...
fix_style(['poster', 'origin'])
Or, assuming you have the Latin Modern Math font installed:
set_style(['origin', 'latex'])
...
fix_style(['origin', 'latex'])
Run the test() routines in publib.test for more examples.
The publib.tools module include independant functions to fix some common matplotlib bugs,
or include extra features. They're usually glanced from somewhere on the web. Proper
referencing is made in the function docstrings.
See for instance:
-
publib.tools.reset_defaults: reset Matplotlib defaults -
publib.tools.regenerate_fonts: rebuild Matplotlib font cache -
publib.tools.fix_bold_TimesNewRoman: fix Times New Roman font appearing bold. See StackOverflow -
publib.tools.keep_color: apply the same color for the next graph to plot -
publib.tools.get_next_color: see which color will be applied next in the color cycle state
plt.plot(...)
keep_color()
plt.plot(...)
publib.tools.list_font_names: to list all fonts available in Python.
See tools.py for more details
-
0.2.2: added tools
-
0.1.9: added talk and OriginPro style
-
0.1.7 : default fonts to Times in article
Some other interesting packages to make nice graphs in Matplotlib.
Add new features:
- pypdfplot: embed the script within the pdf. One single file for the figure and the code!
- brokenaxes
- matplotlib-tools: toolbar (ruler, etc.)
Style based:
- the Matplotlib default style feature
- seaborn
- prettyplotlib
- garrettj403's matplotlib styles for ThesisPlot
Tips and demos of Matplotlib: