python - ecdf 中的彩色线条

Question

我正在尝试为 ecdf 中的一些线条着色，这是我尝试过的：

p = (
    ggplot(data, aes(color='c', x='x')) 
  + stat_ecdf()
).draw()

p = (
    ggplot(data)
  + geom_line(aes(color='c',x='x'), stat='ecdf')
).draw()

在这两种情况下，结果都会忽略颜色参数并仅绘制黑色 cdf。

如果我手动拆分数据，c我可以绘制线条并为其着色，但是我无法弄清楚如何让图例在这种情况下工作。

标签： pythonplotnine

Answer 1

在绘图之前，您需要将颜色“c”列转换为对象类型（因子）。相同的绘图代码应该可以工作。

data['c'] = data['c'].astype(object)