使用 matplolitb 绘制 class 概率

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

使用 matplolitb 绘制 class 概率

Plot class probabilities using matplolitb

python

numpy

matplotlib

scatter-plot