绘制数字线和颜色的数据

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

您可以使用seaborn library＆＃39; s sns.lmplot。查看group的{​​{1}}属性。休息时应遵循评论。

df

Answer 2

使用散点图，您可以提供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()