枚举 matplotlib 图中的图 [英] Enumerate plots in matplotlib figure

问题描述

在matplotlib图中，我想用a)，b)，c)列举所有(子)图，依此类推.有没有办法自动执行此操作?

In a matplotlib figure I would like to enumerate all (sub)plots with a), b), c) and so on. Is there a way to do this automatically?

到目前为止，我使用了各个图的标题，但这远非理想，因为我希望数字左对齐，而可选的真实标题应以图形为中心.

So far I use the individual plots' titles, but that is far from ideal as I want the number to be left aligned, while an optional real title should be centered on the figure.

推荐答案

import string
from itertools import cycle
from six.moves import zip

def label_axes(fig, labels=None, loc=None, **kwargs):
    """
    Walks through axes and labels each.

    kwargs are collected and passed to `annotate`

    Parameters
    ----------
    fig : Figure
         Figure object to work on

    labels : iterable or None
        iterable of strings to use to label the axes.
        If None, lower case letters are used.

    loc : len=2 tuple of floats
        Where to put the label in axes-fraction units
    """
    if labels is None:
        labels = string.ascii_lowercase

    # re-use labels rather than stop labeling
    labels = cycle(labels)
    if loc is None:
        loc = (.9, .9)
    for ax, lab in zip(fig.axes, labels):
        ax.annotate(lab, xy=loc,
                    xycoords='axes fraction',
                    **kwargs)

示例用法:

from matplotlib import pyplot as plt
fig, ax_lst = plt.subplots(3, 3)
label_axes(fig, ha='right')
plt.draw()

fig, ax_lst = plt.subplots(3, 3)
label_axes(fig, ha='left')
plt.draw()

这对我来说似乎很有用，我把它放在了一个要点中:https://gist.github.com/tacaswell/9643166

This seems useful enough to me that I put this in a gist : https://gist.github.com/tacaswell/9643166

这篇关于枚举 matplotlib 图中的图的文章就介绍到这了，希望我们推荐的答案对大家有所帮助，也希望大家多多支持IT屋！