How to save matplotlib plot with the same file name as its filename.py

Question

I run the python code for Matplotlib plotting from thisfilename.py. Is there a variable that stores this filename without file path? As I want to save the plot as thisfilename.png without typing the file name like this every time?

if __name__ == '__main__':
  ...
  plt.savefig('thisfilename.png')

I want to change the last line to something like

  # pseudocode
  plt.savefig(variable, '.png')

I tested

  plt.savefig(__file__, ".png")

The saved filename had .py, the file path, and the space before .png which I don't want

Check out this question

Tom de Geus
– Tom de Geus

2017-06-06 07:19:31 +00:00
Commented Jun 6, 2017 at 7:19 — Tom de Geus
– Tom de Geus, Commented Jun 6, 2017 at 7:19

user2660966 · Accepted Answer · 2017-06-06 07:33:32Z

2

To save the figure with same name as the python file, replacing the ".py" by ".png" and removing the path, I would use :

plt.gcf().savefig(__file__.rstrip('.py').split('/')[-1] + ".png")

the rstrip removes ".py" from the right side of the file string, the split method separate your chains at each "/" and [-1] keeps only the last item of the list the + can be used to concatenate strings

answered Jun 6, 2017 at 7:33

user2660966

9759 silver badges21 bronze badges

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

Comments

MaxPowers · Accepted Answer · 2017-06-06 07:29:50Z

1

import inspect
import matplotlib.pyplot as plt
import numpy as np

frame = inspect.currentframe()
path = inspect.getfile(frame)
fname = path.split('.')[0]

t = np.linspace(0,1, 1000)
plt.plot(t, np.sin(2*np.pi*t*4))
plt.savefig(fname + ".jpg")

answered Jun 6, 2017 at 7:29

MaxPowers

5,5292 gold badges50 silver badges70 bronze badges

Comments

Martin Evans · Accepted Answer · 2017-06-06 07:50:09Z

1

You can use Python's splitext() to break __file__ into two parts and then add the required extension as follows:

import os

output_filename = os.path.splitext(__file__)[0] + '.png'
plt.savefig(output_filename)

answered Jun 6, 2017 at 7:50

Martin Evans

46.9k17 gold badges88 silver badges104 bronze badges

Collectives™ on Stack Overflow

How to save matplotlib plot with the same file name as its filename.py

3 Answers 3

Comments

Comments

Comments

Your Answer

Linked

Hot Network Questions

Collectives™ on Stack Overflow

3 Answers 3

Comments

Comments

Comments

Your Answer

Sign up or log in

Post as a guest

Linked

Related