python - 如何用 True、False 和颜色编码 Python 绘图来标记 Y 轴？

标签 python matplotlib seaborn categories anomaly-detection

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

我已经使用 True 和 False 值绘制 pandas 系列，这些值在下图中自动显示为 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')

Plot Results

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

最佳答案

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

这会产生以下图像

CategoricalPlot