matplolitb를 사용한 플롯 클래스 확률

Question

값이 두 개인 numpy 배열 y_prob과 dataY이 있습니다. dataY은 각 값이 1 또는 0 인 1 차원 배열입니다. y_prob은 2 차원 배열입니다. 위치를 결정하기 위해 y_prob을 사용하고 점의 색을 결정하기 위해 dataY을 사용하여 산점도를 플롯하려합니다. 어떻게해야합니까?
샘플 데이터 :

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

당신은 표준 패키지 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() 

을 사용할 수 있습니다