在 plotnine 中绘制多个 lambda 函数以便出现图例的正确方法是什么？

Question

看起来与我正在寻找的答案相当接近，但还不完全是。也许我没有遵循 "tidy data" 原则，需要在许多点上评估数据框来绘制这些函数，但我犹豫是否接受它作为答案。

这是绘制图表的代码。

call_value_per_unit = lambda s_t1,X: max(0, s_t1-X)
put_value_per_unit = lambda s_t1, X: max(0, X-s_t1)
put_call_value = lambda s_t1, X: put_value_per_unit(s_t1, X) + call_value_per_unit(s_t1, X)

independent_variable = "Stock Price"
dependent_variable = "Asset Price"

g = ggplot(pd.DataFrame({independent_variable:[10,20]}), aes(x=independent_variable)) \
         + stat_function(fun=put_value_per_unit, args=[15], color="red") \
         + stat_function(fun=call_value_per_unit, args=[15], color="blue") \
         + stat_function(fun=put_call_value, args=[15], color="black") \
         + ylab(dependent_variable) \
         + ggtitle(" ".join([independent_variable , "vs", dependent_variable]))
_ = g.draw()

但是没有传说...我希望有一个。

（虽然我在python，R用户可能会有好的建议）

Answer 1

在 R 中，要使用包 ggplot2 绘制函数，首先使用 x 向量定义数据集。然后使用 stat_function 和适当的 geom。这通常是

之一

geom = "line"
geom = "point"

那么，绘制函数图就很简单了。

library(ggplot2)

df1 <- data.frame(x = -5:5)

ggplot(df1, aes(x)) +
  stat_function(geom = "line", fun = function(x) x^2)

Answer 2

不可能在一个图中绘制多个函数并为每个函数添加图例。对于 stat_function，func 是一个非美学参数，因此您不能将 variable/column 映射到它。图例仅有助于解释美学映射。

由于您想进行大量计算，因此请在绘图调用之外执行此操作，然后使用 geom_line 绘制结果。确保您的数据框采用 tidy data 形式。当它不是完成工作的最佳工具时，不要让 stat_function 强迫您使用它。

在 plotnine 中绘制多个 lambda 函数以便出现图例的正确方法是什么？

What is the proper way plot multiple lambda function in plotnine so legend appears?

python

r

ggplot2

plotnine