python - plotnine 直接颜色分配

Question

我不确定这是 plotnine 中的功能，但我想专门为值分配颜色。

例如，如果我正在创建一个散点图

colors = {True:'#a54f7e', False:'grey'}  

(ggplot(df, aes(x = "col1", y="col2", color = "col3")) +
     geom_jitter(size = 3, color = colors)

所以在这种情况下，col3 将是 True/False 值，理想情况下我希望所有 True 值为 #a54f7e，所有 False 值为灰色。上面的代码没有运行并出现此错误:

PlotnineError: "'{True: 'red', False: 'blue'}' does not look like a valid value for `color`"

帮助!

Answer 1

如果你想分配颜色使用scale_color_manual:

import pandas as pd
from plotnine import *

data = [[1, 4.5,True], [2, 4.25,False], [3, 3.75,False],[4, 3.5,True],[1, 4.0,False],[2, 3.75,False],[3, 4.0,False],[4, 4.25,True]]

df = pd.DataFrame(data, columns = ['col1', 'col2','col3'])

colors = {True:'#a54f7e', False:'grey'}  

(ggplot(df, aes(x = "col1", y="col2", color = "col3")) +
     geom_jitter(size = 3) +
     scale_color_manual(values = colors))