Save all open Matplotlib figures
Project description
savefigs
Easily save all open Matplotlib figures, with useful filenames.
Usage
Assume we have a script some_script.py
that creates multiple Matplotlib figures.
Import the savefigs
function:
from savefigs import savefigs
The below examples assume the figures do not have labels (fig.get_label()
).
If a figure does have a label, it will be used in place of fig{num}
.
Default save settings (./{script filename stem}{figure label or fig{num}}.png
):
savefigs()
# ./some_script_fig1.png, ./some_script_fig2.png, ...
👆 The filenames tell us which script generated the figures as well as their relative places in the figure generation order (or labels if they are labeled).
Specify directory:
savefigs(save_dir="figs") # must exist
# ./figs/some_script_fig1.png, ./figs/some_script_fig2.png, ...
Specify a different prefix to the base stem format:
savefigs(stem_prefix="run1")
# ./run1_fig1.png, ./run1_fig2.png, ...
Save in multiple file formats:
savefigs(formats=["png", "pdf"])
# ./some_script_fig1.png, ./some_script_fig1.pdf, ...
Avoid overwriting files:
savefigs(clobber=False, noclobber_method="add_num")
# ./some_script_fig3.png (assuming ./some_script_fig{1,2}.png already exist)
👆 By default (without changing noclobber_method
), setting clobber=False
will instead error.
Background
When writing a script that creates multiple figures, I usually label them (usually using the num
argument to plt.figure()
/plt.subplots()
), which makes it easier to find the correct figure window. Then, at the end of the script I write a loop like:
for num in plt.get_fignums():
fig = plt.figure(num)
fig.savefig(f"{fig.get_label()}.pdf", ...)
# Maybe another format...
savefigs()
essentially does this, but is more robust and provides additional features through keyword arguments. And it saves having to write those lines in the script, instead allowing the simple one-liner:
from savefigs import savefigs; savefigs()
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
Built Distribution
Hashes for savefigs-0.2.1-py2.py3-none-any.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | 2ead50439a2d0e3089be0494f8f886ffa9b2cc9de86a6e4c65feda4b5796ca68 |
|
MD5 | 7c17b86f39c4325c40a7e4ad8fb110b3 |
|
BLAKE2b-256 | f6713c80ae0324a0295cb6861b53ddf5eec37386b9baba6aee6a47921250abb2 |