Skip to main content

Convert matplotlib figures into TikZ/PGFPlots

Project description

matplotlib2tikz

CircleCI codecov Code style: black Documentation Status awesome PyPi Version DOI GitHub stars

This is matplotlib2tikz, a Python tool for converting matplotlib figures into PGFPlots (TikZ) figures like

for native inclusion into LaTeX documents.

The output of matplotlib2tikz is in PGFPlots, a LaTeX library that sits on top of TikZ and describes graphs in terms of axes, data etc. Consequently, the output of matplotlib2tikz retains more information, can be more easily understood, and is more easily editable than raw TikZ output. For example, the matplotlib figure

import matplotlib.pyplot as plt
import numpy as np

plt.style.use("ggplot")

t = np.arange(0.0, 2.0, 0.1)
s = np.sin(2 * np.pi * t)
s2 = np.cos(2 * np.pi * t)
plt.plot(t, s, "o-", lw=4.1)
plt.plot(t, s2, "o-", lw=4.1)
plt.xlabel("time (s)")
plt.ylabel("Voltage (mV)")
plt.title("Simple plot $\\frac{\\alpha}{2}$")
plt.grid(True)

from matplotlib2tikz import save as tikz_save

tikz_save("test.tex")

(see above) gives

% This file was created by matplotlib2tikz vx.y.z.
\begin{tikzpicture}

\definecolor{color1}{rgb}{0.203921568627451,0.541176470588235,0.741176470588235}
\definecolor{color0}{rgb}{0.886274509803922,0.290196078431373,0.2}

\begin{axis}[
title={Simple plot $\frac{\alpha}{2}$},
xlabel={time (s)},
ylabel={Voltage (mV)},
xmin=-0.095, xmax=1.995,
ymin=-1.1, ymax=1.1,
tick align=outside,
tick pos=left,
xmajorgrids,
x grid style={white},
ymajorgrids,
y grid style={white},
axis line style={white},
axis background/.style={fill=white!89.803921568627459!black}
]
\addplot [line width=1.64pt, color0, mark=*, mark size=3, mark options={solid}]
table {%
0 0
0.1 0.587785252292473
% [...]
1.9 -0.587785252292473
};
\addplot [line width=1.64pt, color1, mark=*, mark size=3, mark options={solid}]
table {%
0 1
0.1 0.809016994374947
% [...]
1.9 0.809016994374947
};
\end{axis}

\end{tikzpicture}

(Use get_tikz_code() instead of save() if you want the code as a string.)

Tweaking the plot is straightforward and can be done as part of your LaTeX work flow. The fantastic PGFPlots manual contains great examples of how to make your plot look even better.

Of course, not all figures produced by matplotlib can be converted without error. Notably, 3D plot don't work.

Installation

matplotlib2tikz is available from the Python Package Index, so simply type

pip install -U matplotlib2tikz

to install/update.

Usage

  1. Generate your matplotlib plot as usual.

  2. Instead of pyplot.show(), invoke matplotlib2tikz by

    tikz_save('mytikz.tex')
    

    to store the TikZ file as mytikz.tex. Load the library with:

    from matplotlib2tikz import save as tikz_save
    

    Optional: The scripts accepts several options, for example height, width, encoding, and some others. Invoke by

    tikz_save('mytikz.tex', figureheight='4cm', figurewidth='6cm')
    

    Note that height and width must be set large enough; setting it too low may result in a LaTeX compilation failure along the lines of Dimension Too Large or Arithmetic Overflow; see information about these errors in the PGFPlots manual. To specify the dimension of the plot from within the LaTeX document, try

    tikz_save(
        'mytikz.tex',
        figureheight='\\figureheight',
        figurewidth='\\figurewidth'
        )
    

    and in the LaTeX source

    \newlength\figureheight
    \newlength\figurewidth
    \setlength\figureheight{4cm}
    \setlength\figurewidth{6cm}
    \input{mytikz.tex}
    
  3. Add the contents of mytikz.tex into your LaTeX source code; a convenient way of doing so is via \input{/path/to/mytikz.tex}. Also make sure that in the header of your document the packages for PGFPlots and proper Unicode support and are included:

    \usepackage[utf8]{inputenc}
    \usepackage{pgfplots}
    

    Additionally, with LuaLaTeX

    \usepackage{fontspec}
    

    is needed to typeset Unicode characters. Optionally, to use the latest PGFPlots features, insert

    \pgfplotsset{compat=newest}
    

Contributing

If you experience bugs, would like to contribute, have nice examples of what matplotlib2tikz can do, or if you are just looking for more information, then please visit matplotlib2tikz's GitHub page.

Testing

matplotlib2tikz has automatic unit testing to make sure that the software doesn't accidentally get worse over time. In test/, a number of test cases are specified. Those run through matplotlib2tikz and compare the output with a previously stored reference TeX file.

To run the tests, just check out this repository and type

pytest

Distribution

To create a new release

  1. bump the __version__ number,

  2. publish to PyPi and GitHub:

    $ make publish
    

License

matplotlib2tikz is published under the MIT license.

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

matplotlib2tikz-0.7.1.tar.gz (44.9 kB view hashes)

Uploaded Source

Built Distribution

matplotlib2tikz-0.7.1-py2.py3-none-any.whl (36.1 kB view hashes)

Uploaded Python 2 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