2017-04-18 1 views
0

私は2つのnumpy配列y_probdataYを持っています。 dataYは、各値が1または0である1次元配列です。 y_probは2次元配列です。私は、位置を決定するためにy_probを使用し、ポイントの色を決定するためにdataYを使用して散布図をプロットしたいと思います。これどうやってするの?
サンプルデータ:matplolitbを使ったプロットクラス確率

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

答えて

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() 
標準パッケージを使用することができます
関連する問題