使用 matplolitb 绘制 class 概率
Plot class probabilities using matplolitb
我有两个 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]
您可以使用标准包 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()
我有两个 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]
您可以使用标准包 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()