使用matplolitb绘制类概率

Question

我有两个数值对应的numpy数组y_prob和dataY。 dataY是一维数组，其中每个值都是1或0。 y_prob是一个二维数组。我希望使用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＆amp; 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()