如何基于python中的条件添加geom_point图层

如何解决如何基于python中的条件添加geom_point图层

我正在使用基于R的plotnine的python软件包ggplot，并尝试创建一个绘图，在该绘图中我想绘制一条线，然后，我想添加一个geom_point图层来绘制如果验证了给定的条件，则只有一些要点。

因此，我有一个数据框，其中的变量包含实数（这是我要绘制为一行的实数），然后是二进制变量。如果第二个变量的值为1，我想用第一个变量的值绘制点，而第二个变量的值为0，则什么也不要。

我以一个小数据框为例

import pandas as pd
from plotnine import ggplot,aes,geom_line,geom_point
d = dict(
    a = np.arange(10),b = [0,1,0]
)
d = pd.DataFrame(d)

(ggplot(df,aes(x='a')) + 
 geom_line(aes(y='a'),color='black')
 # Some code to plot points with value a[i] if b[i]==1
 )

我能想到的唯一方法是创建一个第二个数据帧，如果b = 1，则值为b，如果b = 0，则为None，但这有点笨拙

解决方法

您必须将数据框与所需的点图层中的选定数据一起使用。

(ggplot(df,aes(x='a'))
 + geom_line(aes(y='a'),color='black')
 + geom_point(df[df['b'] == 1],aes(y='a')) # or geom_point(df.query('b == 1'),aes(y='a'))
 )