Plot Klasse Wahrscheinlichkeiten mit Matplolitb

Question

Ich habe zwei numpy Arrays y_prob und dataY deren Werte entsprechen. dataY ist ein eindimensionales Array, wobei jeder Wert ein 1 oder ein 0 ist. y_prob ist ein zweidimensionales Array. Ich möchte ein Streudiagramm mit y_prob zur Bestimmung der Position und dataY zur Bestimmung der Farbe des Punktes plotten. Wie kann ich das machen?
Beispieldaten:

y_prob = [[0.5,0.5], [0.3,0.7], [0.2,0.8], [0.1,0.9]] 
dataY = [1,0,0,0] 

Answer 1

Sie die Standardpakete numpy & matplotlib

import numpy as np 
import matplotlib.pyplot as plt 

y_prob = np.array([[0.5,0.5], [0.3,0.7], [0.2,0.8], [0.1,0.9]]) 
dataY = [1,0,0,0] 

fig = plt.figure() 
plt.scatter(x=y_prob[:,0], y=y_prob[:,1], c=dataY) 
fig.show() 

verwenden können