为matplotlib子图设置多个标题

时间:2017-12-12 16:29:23

标签: python matplotlib venn-diagram matplotlib-venn

我正在创建通过matplotlib_venn制作的维恩图的子图。这是我正在制作的情节:

Subplots missing top-right title

注意右上图中缺少的标题。这是因为每当我调用ax.set_title时,它都会替换现有的子图标题。创建此图的代码是:

oadoi_colors = ['all', 'closed', 'closed + green', 'bronze', 'green', 'hybrid', 'gold']
collections = ['Combined', 'Crossref', 'Unpaywall', 'Web of Science']
figure, axes = matplotlib.pyplot.subplots(nrows=len(oadoi_colors), ncols=len(collections), figsize=(2.7 * len(collections), 2.7 * len(oadoi_colors)))
subplot_label_rectprops = {'facecolor': '#fef2e2', 'boxstyle': 'round'}

for (oadoi_color, collection), df in coverage_df.groupby(['oadoi_color', 'collection']):
    i = oadoi_colors.index(oadoi_color)
    j = collections.index(collection)
    ax = axes[i, j]
    venn_plotter(df, ax=ax)
    if i == 0:
        text = ax.set_title(collection, loc='center', y=1.1)
        text.set_bbox(subplot_label_rectprops)
        # Top-right subplot cannot titled twice: https://stackoverflow.com/questions/36602347
    if j == len(collections) - 1:
        text = ax.set_title(oadoi_color.title(), x=1.1, y=0.5, verticalalignment='center', rotation=270)
        text.set_bbox(subplot_label_rectprops)

venn_plotter是一个调用matplotlib_venn.venn3_unweighted的函数。 dfpandas.DataFrame数据。

在matplotlib title demo中,您似乎可以设置多个标题......但我无法弄清楚如何使用子图进行此操作。

answer to a similar question建议使用set_ylabel。但是,设置ylabel对这些静脉图没有影响。

1 个答案:

答案 0 :(得分:3)

问题是每个轴子图都只有一个标题。如果您再次拨打set_title,则会替换第一个标题 标题和文本之间没有太大区别。因此,您可以决定对行的列和文本使用标题。

import matplotlib.pyplot as plt

fig, axes = plt.subplots(3,3)
for i in range(3):
    for j in range(3):
        ax = axes[i,j]
        if i == 0:
            title = ax.set_title("column title", loc='center', y=1.1)
        if j == 2:
            text = ax.text(1.1,0.5,"row title", size=12,
                           verticalalignment='center', rotation=270)

plt.show()

enter image description here