Как сгенерировать ключ в matplotlib

Question

Я строю несколько трехмерных сфер в matplotlib, и каждая из них будет помечена как 1,2,3, и так далее.Я делаю это с помощью ax.text ().Однако каждое из этих чисел соответствует имени, и я хочу, чтобы ключ этих данных отображался рядом с графиком после его генерации;как бы мне поступить об этом?

Пока мой код:

import matplotlib.pyplot as plt
from mpl_toolkits.mplot3d import Axes3D
import numpy

fig = plt.figure()
ax = fig.gca(projection='3d')
ax.set_aspect('equal')

u = numpy.linspace(0, 2*numpy.pi, 50)
v = numpy.linspace(0, numpy.pi, 50)
x = numpy.outer(numpy.cos(u), numpy.sin(v))
y = numpy.outer(numpy.sin(u), numpy.sin(v))
z = numpy.outer(numpy.ones(numpy.size(u)), numpy.cos(v))

for k in range(1):
    c = [0,0,0]
    ax.text(c[0],c[1],c[2],"1")
    r = .1
    ax.plot_surface(
        r*x + c[0], r*y + c[1], r*z + c[2],
        color='#1f77b4',
        alpha=0.5,
        linewidth=0
        )

plt.show()

Я хочу, чтобы метка "1" соответствовала "Имя";оба они будут из списков, но это всего лишь пример!