Skip to main content

rsmf (right-size my figures) helps you prepare publication-ready figures with matplotlib.

Project description

Documentation Status

rsmf (right-size my figures)

When I am writing a paper I am a bit picky about the figures. It is especially important for me that the fonts and font sizes match the surrounding document. As I usually plot with matplotlib I created this library to help with that. This library provides a means to automatically adjust figure sizes and font sizes in matplotlib to fit the ones in commonly used scientific journals. Currently quantumarticle and revtex4-1 are supported.

Install

Currently, you have to install the package from GitHub.

pip install git+https://www.github.com/johannesjmeyer/rsmf

But after the alpha phase is over it will end up on PyPI!

Usage

Setup

You need to tell rsmf how you set up your document by invoking rsmf.setup. This can be done in two ways. Either, you give rsmf the \documentclass string used for setting up the document, as in

import rsmf
formatter = rsmf.setup(r"\documentclass[a4paper,12pt,noarxiv]{quantumarticle}")

The r in front of the string is necessary so that \d is not mistaken for an escape sequence. If you have your document stored locally, there is an even more convenient way: you can just supply rsmf with the path to your main tex file (the one containing the document setup) and it will find that out for itself:

formatter = rsmf.setup("example.tex")

This is especially cool because rsmf will automatically adjust the plots when the underlying document class is changed without any needs to change python code! This makes swapping journals a lot easier.

Figures

The setup routine will return a formatter. This formatter can then be used to create matplotlib figure objects by invoking the method formatter.figure. It has three arguments:

  • aspect_ratio (float, optional): the aspect ratio (width/height) of your plot. Defaults to the golden ratio.
  • width_ratio (float, optional): the width of your plot in multiples of \columnwidth. Defaults to 1.0.
  • wide (bool, optional): indicates if the figures spans two columns in twocolumn mode, i.e. if the figure* environment is used, has no effect in onecolumn mode . Defaults to False.

This is the place where you set the width of your plots, not in the LaTeX document. If you include the resulting figure with a different width, the font sizes will not match the surrounding document!

For example, a regular figure is created via

fig = formatter.figure(aspect_ratio=.5)

# ... some plotting ...
plt.savefig("example.pdf")

and included via

\begin{figure}
	\centering
	\includegraphics{example}
	\caption{...}
\end{figure}

A wide figure that spans 80% of the page on the other hand is created by

fig = formatter.figure(width_ratio=.8, wide=True)

# ... some plotting ...
plt.savefig("example_wide.pdf")

and included via the multi-column figure* environment:

\begin{figure*}
	\centering
	\includegraphics{example_wide}
	\caption{...}
\end{figure*}

Note that you should always save your figures in some sort of vectorized format, like pdf and that calling plt.tight_layout() before saving usually makes your plots nicer.

Other features

You can access the underlying fontsizes via formatter.fontsizes. The nomenclature follows that of LaTeX itself, so we have

formatter.fontsizes.tiny
formatter.fontsizes.scriptsize
formatter.fontsizes.footnotesize
formatter.fontsizes.small
formatter.fontsizes.normalsize
formatter.fontsizes.large
formatter.fontsizes.Large
formatter.fontsizes.LARGE
formatter.fontsizes.huge
formatter.fontsizes.Huge

This is especially useful if you want to tweak titles, legends and annotations while still having proper (LaTeX) fontsizes.

Using rsmf with other frameworks

You can use rsmf together with your favorite plotting framework, for example seaborn. There is only one catch: if you use matplotlib styles or seaborn styles, you might overwrite the settings imposed by rsmf, especially regarding font-size. To this end, the formatters have a method formatter.set_default_fontsizes that only change the underlying fontsizes. An example use would be

fig = formatter.figure(wide=True)
sns.set(style="ticks")
formatter.set_default_fontsizes()

# ... some plotting ...

Sometimes these styles also overwrite other things, like the font family (serif/sans-serif). There is no correction method for that yet.

Example

An example document alongside with a notebook for making the plots used can be found in the examples folder.

How it works

Under the hood, rsmf sets the matplotlib backend to pgf, which allows the use of LaTeX. For each supported document class, the specific column widths and font sizes are stored in code, alongside with packages that are loaded that change fonts. For quantumarticle, for example, the package lmodern is loaded into the pgf backend to get the right sans-serif font.

When calling rsmf.setup, matplotlib's rcParams are adjusted to make the fontsizes match the surrounding document. Note that formatter.figure does not mess with rcParams.

Contribute

Do you have trouble setting up plots for your favorite document class and it is not supported here? Do not hesitate to make a PR!

Project details


Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Source Distribution

rsmf-0.1.tar.gz (6.1 kB view hashes)

Uploaded Source

Built Distribution

rsmf-0.1-py3-none-any.whl (8.1 kB view hashes)

Uploaded Python 3

Supported by

AWS AWS Cloud computing and Security Sponsor Datadog Datadog Monitoring Fastly Fastly CDN Google Google Download Analytics Microsoft Microsoft PSF Sponsor Pingdom Pingdom Monitoring Sentry Sentry Error logging StatusPage StatusPage Status page