Savefig with differents names inside a for loop. Python

Question

Im calculating the flow in a lid-driven cavity, and Im plotting the result with a quiver. I want to save the plot in every time step, but obviously, as the name is the same, it´s only keep the last one, how can I do it?

for n in range(nt):

   #Here I do all the calculation to obtain the new u and v

   uC=0.5*(u[:,1:] + u[:,:-1])
   vC=0.5*(v[1:,:] + v[:-1,:])

   plt.cla()
   plt.quiver(x, y, uC, vC);
   plt.draw()
   plt.savefig( "Instant1.png")

So, imagine nt = 10, I want to get ten differents png files. Any ideas? I aprecciate all your help

Kevin · Accepted Answer · 2015-10-02 14:24:02Z

3

You could change the file name each time:

plt.savefig("Instant{}.png".format(n))

Also, If you have more than ten plots, it might be a good idea to have some leading zeroes, eg. so "Instant5.png" doesn't come after "Instant10.png" in lexicographic order.

plt.savefig("Instant{:03}.png".format(n))

answered Oct 2, 2015 at 14:24

Kevin

76.5k13 gold badges141 silver badges168 bronze badges

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

1 Comment

Fernando Bastos García Over a year ago

Just what I needed. Thanks!

dsh · Accepted Answer · 2015-10-02 20:01:38Z

3

You could also do:

plt.savefig("Instand"+str(n)+".png")

edited Oct 2, 2015 at 20:01

dsh

12.3k3 gold badges37 silver badges53 bronze badges

answered Oct 2, 2015 at 14:43

augre

3503 silver badges13 bronze badges

Collectives™ on Stack Overflow

Savefig with differents names inside a for loop. Python

2 Answers 2

1 Comment

Comments

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

2 Answers 2

1 Comment

Comments

Your Answer

Sign up or log in

Post as a guest

Related