seaborn distplot loop lazy evaluation

Question

I am using ipython notebook and trying to use the following function to export seaborn distplots. It works just fine if I call the function and execute with only one variable at a time. If I call the function in a loop, it continues to build on top of the distplot from the previous function call.

My desired output would be for the function to output a new displot every time it is called in a loop. Is there a way to force evaluation or a new distplot?

def graph_extraversion (x):


    file_name = "extraversion_" + str(x) + ".png"
    sns_plot = sns.distplot(Personalities[Personalities.labels1 ==x].extraversion)
    sns_plot = sns.distplot(df.extraversion)
    fig = sns_plot.get_figure()
    fig.savefig(file_name)
    new_stat = Personalities[Personalities.labels1 ==x].extraversion.describe()
    extraversion_drift = extraversion_median - new_stat[1]
    drift = extraversion_drift / extraversion_std
    if (drift >= 1) | (drift <= -1):
        return "1 std deviation or more"
    else:
        return "Less than one std deviation"

This what is what the distplot looks like after one call

This is two calls later in a loop.

Again this works just fine with a single call and execution but when looped it keeps building.

You may want to look at stackoverflow.com/questions/23969619/…. — mwaskom
– mwaskom, Commented Jul 11, 2016 at 11:52
I looked at the posting, but is there a way to return the function to the caller and instantiate a new figure in a loop. — Benny Baysinger
– Benny Baysinger, Commented Jul 14, 2016 at 1:54

Benny Baysinger · Accepted Answer · 2016-07-28 02:18:01Z

1

So this has to do with matplotlib and closing figures.

additional code required is an import:

import matplotlib.pyplot as plt

Then at the end of the func:

plt.close(fig)

This should help with any looping with both seaborn and matplotlib

answered Jul 28, 2016 at 2:18

Benny Baysinger

701 silver badge10 bronze badges

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

1 Comment

pbreach Over a year ago

No additional import needed, just fig.close(). Using the figure object seems to be the more acceptable way to do this.

Collectives™ on Stack Overflow

seaborn distplot loop lazy evaluation

1 Answer 1

1 Comment

Your Answer

Linked

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

1 Comment

Your Answer

Sign up or log in

Post as a guest

Linked

Related