barre de couleurs matplotlib pour scatter

Je travaille avec des données dont les données ont 3 paramètres de traçage: x, y, c. Comment créer une valeur de couleur personnalisée pour un nuage de points?

Prolonger cet exemple que j'essaye de faire:

import matplotlib
import matplotlib.pyplot as plt
cm = matplotlib.cm.get_cmap('RdYlBu')
colors=[cm(1.*i/20) for i in range(20)]
xy = range(20)
plt.subplot(111)
colorlist=[colors[x/2] for x in xy] #actually some other non-linear relationship
plt.scatter(xy, xy, c=colorlist, s=35, vmin=0, vmax=20)
plt.colorbar()
plt.show()

mais le résultat est TypeError: You must first set_array for mappable

À partir de la documentation matplotlib sur le scatter 1 :

cmap n'est utilisé que si c est un tableau de flottants

La liste de couleurs doit donc être une liste de flottants plutôt qu'une liste de tuples comme vous l'avez maintenant. plt.colorbar () veut un objet mappable, comme le CircleCollection que plt.scatter () renvoie. vmin et vmax peuvent alors contrôler les limites de votre barre de couleurs. Les choses en dehors de vmin / vmax obtiennent les couleurs des points de terminaison.

Comment cela fonctionne-t-il pour vous?

import matplotlib.pyplot as plt
cm = plt.cm.get_cmap('RdYlBu')
xy = range(20)
z = xy
sc = plt.scatter(xy, xy, c=z, vmin=0, vmax=20, s=35, cmap=cm)
plt.colorbar(sc)
plt.show()

Voici la façon OOP d'ajouter une barre de couleurs:

fig, ax = plt.subplots()
im = ax.scatter(x, y, c=c)
fig.colorbar(im, ax=ax)

Si vous cherchez à disperser par deux variables et la couleur par la troisième, Altair peut être un excellent choix.

Création du jeu de données

import matplotlib.pyplot as plt
import numpy as np
import pandas as pd

df = pd.DataFrame(40*np.random.randn(10, 3), columns=['A', 'B','C'])

Parcelle Altair

from altair import *
Chart(df).mark_circle().encode(x='A',y='B', color='C').configure_cell(width=200, height=150)

Terrain