在数轴和颜色上绘制数据

Question

我有两个列表predictedY，其中包含n从 0 到 1 的元素，并且unlabelledY包含n元素 1 或 0。

predictedY我希望在数轴上绘制 的值，如果 中的相应元素unlabeledY是 1，则将其着色为红色，否则将其着色为黑色。

我怎样才能做到这一点？

predictedY =[0.456,0.962,0.231]
UnlabelledY=[0,1,0]

Answer 1

使用散点图，您可以提供UnlabelledY颜色参数。选择合适的颜色图（在本例中例如"hot_r"使用vmin=-2）可以让您获得您选择的颜色。

import numpy as np
import matplotlib.pyplot as plt

predictedY =[0.456,0.962,0.231] 
UnlabelledY=[0,1,0]

plt.scatter(predictedY, np.zeros_like(predictedY), 
            c=UnlabelledY, cmap="hot_r", vmin=-2)

plt.yticks([])
plt.show()