如何使用 plotnine 库在 Python 中绘制函数

Question

我一直是 R 的长期用户，最近过渡到 Python，并且我一直在尝试使用 ggplot2 进行绘图，因为它非常直观。Plotnine被认为是最 ggplot2 风格的绘图库，我已经成功地用它重新创建了大多数图形，除了关键的是如何绘制常规函数。

在基础 R 中，您可以轻松定义一个方程，如so，将结果输入到一个stat_function()层中，并设置图形的限制代替dataarg.，并成功绘制抛物线等。但是，设置图形限制的语法在 Python 中必须不同（也许使用 numpy？），并且使用 定义方程sympy，这对我来说是另一个分歧。

那么我将如何使用 plotnine 绘制函数呢？以上两个障碍是我认为给我带来麻烦的与 ggplot2 的两个不同之处，因为 plotnine 在网上的例子太少了。

PS 这是我想使用 plotnine 在 Python 中重新创建的示例：

> library(ggplot2)
> basic_plot <- function(x) x^2 + 2.5
> graph <- ggplot(data.frame(x=c(-5,5)), aes(x=x)) +
+ stat_function(fun = basic_plot)
> graph

Answer 1

您不需要 numpy，它以“标准”方式工作得很好！:)

from plotnine import *
import pandas as pd

(ggplot(pd.DataFrame(data={"x": [-5, 5]}), aes(x="x"))
    + stat_function(fun=lambda x: x**2+2.5))