Matplotlib subplot title, figure title formatting

Question

How would I go about formatting the below pie chart subplots so that there is more white-space between the fig title and subplot titles. Ideally the subplot titles would also be in closer vicinity to the actual pie chart itself.

I can't seem to find anything in the docs which might enable this, but I'm new to matplotlib.

import matplotlib.pyplot as plt
import pandas as pd
from pandas import DataFrame, Series

m = {"Men" :  {"Yes": 2, "No": 8}}
w = {"Women": {"Yes": 3, "No": 7}}
data = {**m, **w}
df = DataFrame(data)

fig, axes = plt.subplots(1, len(df.columns))
fig.suptitle("Would you prefer to work from home?", fontsize=18)
logging.debug("fig.axes: {}".format(fig.axes))

for i, ax in enumerate(fig.axes):
    col = df.ix[:, i]
    ax = fig.axes[i]
    pcnt_col = col / col.sum() * 100
    ax.set_title("{} (n={})".format(pcnt_col.name, col.sum()))
    ax.pie(pcnt_col.values, labels=pcnt_col.index,
                    autopct="%1.1f%%", startangle=90)
    ax.axis("equal")
    plt.legend(loc="lower right", title="Answer", fancybox=True,
               ncol=1, shadow=True)
plt.show()

piRSquared · Accepted Answer · 2017-01-30 07:54:32Z

7

Use subplots_adjust to separate the two

plt.subplots_adjust(top=0.75)

import matplotlib.pyplot as plt
import pandas as pd
from pandas import DataFrame, Series

m = {"Men" :  {"Yes": 2, "No": 8}}
w = {"Women": {"Yes": 3, "No": 7}}
data = {**m, **w}
df = DataFrame(data)

fig, axes = plt.subplots(1, len(df.columns))
fig.suptitle("Would you prefer to work from home?", fontsize=18)
logging.debug("fig.axes: {}".format(fig.axes))

for i, ax in enumerate(fig.axes):
    col = df.ix[:, i]
    ax = fig.axes[i]
    pcnt_col = col / col.sum() * 100
    ax.set_title("{} (n={})".format(pcnt_col.name, col.sum()))
    ax.pie(pcnt_col.values, labels=pcnt_col.index,
                    autopct="%1.1f%%", startangle=90)
    ax.axis("equal")
    plt.legend(loc="lower right", title="Answer", fancybox=True,
               ncol=1, shadow=True)
plt.subplots_adjust(top=0.55)
plt.show()

answered Jan 30, 2017 at 7:54

piRSquared

296k68 gold badges509 silver badges654 bronze badges

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

3 Comments

piRSquared Over a year ago

@SiHa milage may vary... legend wasn't in the same position on my screen as the OP with OP code.

SiHa Over a year ago

OK, fair enough :)

Michael Johnson Over a year ago

I used keyword arg 'bbox_to_anchor=(1.2, -0.1)' in plt.legend to move the legend to the bottom right hand corner of the figure.

Collectives™ on Stack Overflow

Matplotlib subplot title, figure title formatting

1 Answer 1

3 Comments

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

3 Comments

Your Answer

Sign up or log in

Post as a guest

Related