L'ajout de légende au-dessous de l'axe des X pour un nuage de points à l'aide de matplotlib

Quelque chose comme:

from matplotlib import pyplot as plt
import numpy as np

txt="I need the caption to be present a little below X-axis"

# make some synthetic data
x = np.linspace(0, 1, 512)
y = np.random.rand(512)*2.3 + .1

fig = plt.figure()
ax1 = fig.add_axes((0.1, 0.2, 0.8, 0.7))

ax1.set_title("This is my title")
ax1.set_xlabel('X-axis')
ax1.set_ylabel('Y-axis')

# make the edge colors match the facecolors
ax1.scatter(x,y, c='r', edgecolors='face')
# center text
fig.text(.5, .05, txt, ha='center')

# use OO interface    
ax1.set_xlim([0, 1.05])
ax1.set_ylim([0, 2.5])

# resize the figure to match the aspect ratio of the Axes    
fig.set_size_inches(7, 8, forward=True)

plt.show()

pourrait fonctionner. Rendre cela plus facile à faire est sur le radar pour la mpl en amont, mais nous sommes toujours à la recherche de quelqu'un pour le faire.

Tout d'abord, je me sens bizarre de poster une réponse contre le co-lead developer de matplotlib. Évidemment, @tacaswell sait matplotlib beaucoup mieux que je ne le sera jamais. Mais dans le même temps, sa réponse n'était pas assez dynamique pour moi. J'avais besoin d'une légende qui serait toujours en fonction de la position de la xlabel, et ne pas simplement utiliser les annotations de texte.

J'ai considéré simplement en changeant le xlabel pour ajouter un saut de ligne et le texte de la légende, mais ce ne serait pas de différencier clairement la légende, et vous ne pouvez pas faire des choses comme changer la taille du texte ou de le faire en italique dans le milieu d'une chaîne de texte.

J'ai résolu ce problème en utilisant matplotlib TeX de capacités. Voici ma solution:

from matplotlib import pyplot as plt
from matplotlib import rc
import numpy as np
from pylab import *

rc('text', usetex=True)

file = open('distribution.txt', 'r')

txt="I need the caption to be present a little below X-axis"

x=[]
y=[]
for line in file:
    new=line.rstrip()
    mystring=new.split("\t")
    x.append(mystring[0])
    y.append(mystring[1])


fig = plt.figure()
ax1 = fig.add_axes((0.1,0.4,0.8,0.5))
ax1.set_title("This is my title")
ax1.set_xlabel(r'\begin{center}X-axis\\*\textit{\small{' + txt + r'}}\end{center}')
ax1.set_ylabel('Y-axis')
ax1.scatter(x,y, c='r')
plt.xlim(0, 1.05)
plt.ylim(0, 2.5)
plt.show()

J'ai fait la même chose avec l'aléatoire nuage de points à partir de tacaswell réponse, et voici mon résultat:

Un avertissement: si vous tordre de prendre une entrée variables de chaîne, les chaînes peuvent ne pas être correctement échappé pour une utilisation avec TeX. S'échapper du code LaTeX est déjà couvert sur un Débordement de Pile à https://stackoverflow.com/a/25875504/1404311 . J'ai utilisé directement, et pourrait ensuite prendre arbitraire xlabels et de légendes.