How to fix error with saving plots in matplotlib?

Question

I have this code:

plt.figure(figsize=(24, 9))
function_names = ['Loss', 'Accuracy']
stats_names = iter(list(stats.keys()))

for i in range(2):
    ax = plt.subplot(1, 2, i+1)
    ax.plot(range(config['n_train_epoch']),
            stats[next(stats_names)], 
            label='Validation', 
            color='darkorchid', 
            lw=2.5)
    ax.plot(range(config['n_train_epoch']), 
            stats[next(stats_names)], 
            label='Training', 
            color='mediumspringgreen', 
            lw=2.5)
    ax.set_xlabel('Number of training epochs')
    ax.set_ylabel(function_names[i] + ' value')
    ax.set_title(function_names[i] + ' Functions', fontsize=20)
    ax.legend(fontsize=14)

And i`m getting this plots.

I want to save it to png, but when i refactor my code to this:


plt.figure(figsize=(24, 9))
function_names = ['Loss', 'Accuracy']
stats_names = iter(list(stats.keys()))

for i in range(2):
    ax = plt.subplot(1, 2, i+1)
    ax.plot(range(config['n_train_epoch']),
            stats[next(stats_names)], 
            label='Validation', 
            color='darkorchid', 
            lw=2.5)
    ax.plot(range(config['n_train_epoch']), 
            stats[next(stats_names)], 
            label='Training', 
            color='mediumspringgreen', 
            lw=2.5)
    ax.set_xlabel('Number of training epochs')
    ax.set_ylabel(function_names[i] + ' value')
    ax.set_title(function_names[i] + ' Functions', fontsize=20)
    ax.legend(fontsize=14)

plt.savefig('results/graphics.png')

i`m getting this

What is the problem?

Possible duplicate stackoverflow.com/questions/9622163/… ? — Vigneswaran C
– Vigneswaran C, Commented May 22, 2020 at 4:40

Marcin · Accepted Answer · 2020-05-22 04:49:13Z

0

Matplotlib provides savefig which supports png and other formats.

You could, for example, do:

plt.gcf().savefig('plot.png', dpi=150)

edited May 22, 2020 at 4:49

answered May 22, 2020 at 4:43

Marcin

241k16 gold badges315 silver badges368 bronze badges

Sign up to request clarification or add additional context in comments.

Comments

O.rka · Accepted Answer · 2020-05-22 16:44:29Z

0

I would recommend instantiating your figures and axes directly. It's much easier to manipulate and save plots.

For example,

with plt.style.context("seaborn-white"):
    fig, ax = plt.subplots()
    # Plot something using `ax`
    fig.savefig("path/to/output.png", dpi=300, bbox_inches="tight")

answered May 22, 2020 at 16:44

O.rka

31.1k78 gold badges213 silver badges336 bronze badges

Collectives™ on Stack Overflow

How to fix error with saving plots in matplotlib?

2 Answers 2

Comments

Comments

Your Answer

Linked

Hot Network Questions

Collectives™ on Stack Overflow

2 Answers 2

Comments

Comments

Your Answer

Sign up or log in

Post as a guest

Linked

Related