Matplotlib Colorbar für Streuung

Ich arbeite mit Daten, deren Daten 3 Plotparameter haben: x, y, c. Wie erstellen Sie einen benutzerdefinierten Farbwert für ein Streudiagramm?

Ich versuche dieses Beispiel zu erweitern:

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()

aber das Ergebnis ist TypeError: You must first set_array for mappable

Aus den Matplotlib-Dokumenten zu Streuung 1 :

cmap wird nur verwendet, wenn c ein Array von Floats ist

Die Farbliste muss also eine Liste von Floats sein und keine Liste von Tupeln, wie Sie sie jetzt haben. plt.colorbar () möchte ein zuordnbares Objekt wie die CircleCollection, die plt.scatter () zurückgibt. vmin und vmax können dann die Grenzen Ihrer Farbleiste steuern. Dinge außerhalb von vmin / vmax erhalten die Farben der Endpunkte.

Wie funktioniert das bei Ihnen?

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()

Hier ist die OOP-Methode zum Hinzufügen einer Farbleiste:

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

Wenn Sie nach zwei Variablen streuen und nach der dritten Farbe streuen möchten , ist Altair eine gute Wahl.

Datensatz erstellen

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'])

Altair Grundstück

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

Handlung