CODEWITHSUNDEEP
pythonmatplotlibjupyter-notebook
siva82kb
siva82kb

Reputation: 1922

How to have a function return a figure in python (using matplotlib)?

Assume that I have some data, and I want to create a plot of this data by passing it to a custom plotting function (myplot()). I am using the matplotlib's modules in myplot().

I would like myplot() to return the handle to a figure, and not plot display the plot when I call this function. Here is a sample code and output from iPython.

enter image description here

I have two questions regarding this:

  1. Why do I still see a plot, even though I am assigning the output of myplot() to f?
  2. What do I need to supress this plot when I am assigning the output of myplot() to a variable?

Upvotes: 13

Views: 50160

Answers (2)

ala
ala

Reputation: 224

If you do not want to start the whole notebook in non-inline-modus you can just use the following code:

%config InlineBackend.close_figures = False

def myplot(t,x):
    fig = figure()
    x = plot(t,x)
    fig.savefig('plot.png') # This is just to show the figure is still generated
    return fig

t = arange(0,6,0.01)
x = sin(t)

f = myplot(t,x)

Upvotes: 1

unutbu
unutbu

Reputation: 879701

Start ipython with

ipython notebook

rather than

ipython notebook --pylab=inline

enter image description here

Upvotes: 8

Related Questions