如何在Python图上用True，False和颜色代码标记Y轴？

Question

我正在尝试绘制来自熊猫系列对象True的数据的False data3值。我希望对True和False点进行颜色编码，并希望Y轴指示True和False。

我已经在绘制具有True和False值的熊猫系列，它们在下图中自动显示为1/0。

   import matplotlib.pyplot as plt
   from matplotlib.pyplot import figure
   figure(num=None, figsize=(20,10), dpi=80, facecolor='w', edgecolor='k')
   plt.scatter(x = data3.index, y= data3, color='g', marker='d')

在标签中包含seaborn，以查看是否有更好的选择。

Answer 1

from matplotlib.pyplot import figure
import matplotlib.pyplot as plt
import matplotlib.colors as clrs
figure(num=None, figsize=(20,10), dpi=80, facecolor='w', edgecolor='k')
cmap = clrs.ListedColormap(['red', 'green'])
plt.yticks([1.0, 0.0], ["True",
                        "False"])
plt.scatter(x = data3.index, y= data3, c=(data3 != True).astype(float), marker='d', cmap=cmap)#plt.cm.get_cmap('RdBu'))

这将产生以下图像