首页 > 解决方案 > 如何用 Python 绘图的 True、False 和颜色代码标记 Y 轴?

问题描述

我正在尝试绘制True False熊猫系列对象中的数据值data3。我希望TrueandFalse点用颜色编码,Y 轴表示Trueand False

我已经在绘制带有TrueFalse值的熊猫系列,这些值在下图中自动显示为 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 看看是否有更好的选择。

标签: pythonmatplotlibseaborncategoriesanomaly-detection

解决方案


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'))

这将产生以下图像

分类图


推荐阅读