Python - plt.subplot() with plot created in a for loop

Question

I create the following plots within a for loop:

cat_var={}
categories=['symboling','fuel-type','make']

for x in categories:
    cat_var[x]=df[x].unique()

for key in cat_var.keys():
    plt.figure(figsize=(10,10))
    for value in cat_var[key]:
        subset= df[df[key] == value]
        sns.distplot(subset['price'], hist = False, kde = True,kde_kws = {'linewidth': 3},label = value) 
    plt.legend(prop={'size': 16}, title = key)
    plt.title('Price distribution per %s level'% key)
    plt.xlabel('Price')
    plt.ylabel('Density')
    plt.show()

It is working, but I would like to plot them so they appear all on the same row, whereas now I have them all on the same column. I was thinking about using something like:

fig, axes = plt.subplots(1, 3,figsize=(20,20))

but this does not produce the desired output wherever I put this last line of code.

Any thoughts?

Yuca · Accepted Answer · 2018-07-24 15:03:07Z

1

If you want to have three plots on the same figure, then you need to use subplots and specify where each plot goes

fig, ax = plt.subplots(1, 3,figsize=(20,20))
for idx, key in enumerate(cat_var):       
    for value in cat_var[key]:
        subset= df[df[key] == value]
        sns.distplot(subset['price'], hist = False, ax= ax[idx], kde = True,kde_kws = {'linewidth': 3},label = value)  
plt.show()

With the following output:

edited Jul 24, 2018 at 15:03

answered Jul 24, 2018 at 14:04

Yuca

6,1114 gold badges26 silver badges45 bronze badges

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

6 Comments

Yuca Over a year ago

it's really hard to provide a working solution without your data, but it is a good starting point :)

mcpj Over a year ago

Here is where the data can be found: archive.ics.uci.edu/ml/machine-learning-databases/autos/… you are right I should have included that in my question. The solution is not really working yet, but that is definitely a good start.

Yuca Over a year ago

where are the headers? your data is non labeled hence the column syntax you're using can't be used on the raw data

Yuca Over a year ago

I just tried my code with dummy column names and it worked

Yuca Over a year ago

that's because plt is used when you're doing a simple plot. When you're doing subplots you need to use ax[idx].set_title('your title')

|

Collectives™ on Stack Overflow

Python - plt.subplot() with plot created in a for loop

1 Answer 1

6 Comments

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

6 Comments

Your Answer

Sign up or log in

Post as a guest

Related