使用 add_lines() 和 fitted(lm(y~x)) 围绕 plot_ly() 自定义 R 函数

Question

我想在 R 中围绕 plot_ly() 编写一个自定义函数。这样，我可以制作一系列具有相同格式和样式的散点图，但不会重复代码。 I used this page as a guide. 此代码重现错误：

library(plotly)
my_plot <- function(x, y, ...) {
  require(broom)
  plot_ly(data = mtcars, y = y, x = x, showlegend = FALSE, ...) %>%
    add_markers(y = y) %>%
    add_lines(y = ~fitted(lm(y ~ x))) %>%
    add_ribbons(data = augment(lm(y ~ x, data = mtcars)),
                ymin = ~.fitted - 1.96 * .se.fit,
                ymax = ~.fitted + 1.96 * .se.fit,
                line = list(color = 'rgba(7, 164, 181, 0.05)'),
                fillcolor = 'rgba(7, 164, 181, 0.2)',
                name = "Standard Error")
}
my_plot(y = ~mpg, x = ~disp)

问题行是：

add_lines(y = ~fitted(lm(y ~ x))) %>%

我尝试使用 as.formula()，但错误消息类似。

add_lines(y = ~fitted(lm(as.formula("y ~ x"))) %>%

错误信息如下：

Error in model.frame.default(formula = y ~ x, data = mtcars, drop.unused.levels = TRUE) : object is not a matrix

代码在不是函数时有效：

library(plotly)
library(broom)
plot_ly(data = mtcars, y = ~mpg, x = ~disp, showlegend = FALSE) %>%
  add_markers(y = ~mpg) %>%
  add_lines(y = ~fitted(lm(mpg ~ disp))) %>%
  add_ribbons(data = augment(lm(mpg ~ disp, data = mtcars)),
            ymin = ~.fitted - 1.96 * .se.fit,
            ymax = ~.fitted + 1.96 * .se.fit,
            line = list(color = 'rgba(7, 164, 181, 0.05)'),
            fillcolor = 'rgba(7, 164, 181, 0.2)',
            name = "Standard Error")

Answer 1

一种可能性是将数据框和列名都传递给您的函数，例如

library(plotly)

my_plot <- function(data, x, y, ...) {
  require(broom)
  plot_ly(y = data[[y]], x = data[[x]], showlegend = FALSE, ...) %>%
    add_markers(y = data[[y]]) %>%
    add_lines(y = ~fitted(lm(data[[y]] ~ data[[x]]))) %>%
    add_ribbons(data = augment(lm(data[[y]] ~ data[[x]])),
                ymin = ~.fitted - 1.96 * .se.fit,
                ymax = ~.fitted + 1.96 * .se.fit,
                line = list(color = 'rgba(7, 164, 181, 0.05)'),
                fillcolor = 'rgba(7, 164, 181, 0.2)',
                name = "Standard Error")
}
my_plot(data = mtcars, y = 'mpg', x = 'disp')

或者只是列本身。

library(plotly)

my_plot <- function(x, y, ...) {
  require(broom)
  plot_ly(y = y, x = x, showlegend = FALSE, ...) %>%
    add_markers(y = y) %>%
    add_lines(y = ~fitted(lm(y ~ x))) %>%
    add_ribbons(data = augment(lm(y ~ x)),
                ymin = ~.fitted - 1.96 * .se.fit,
                ymax = ~.fitted + 1.96 * .se.fit,
                line = list(color = 'rgba(7, 164, 181, 0.05)'),
                fillcolor = 'rgba(7, 164, 181, 0.2)',
                name = "Standard Error")
}
my_plot(y = mtcars$mpg, x = mtcars$disp)

使用 add_lines() 和 fitted(lm(y~x)) 围绕 plot_ly() 自定义 R 函数

Custom R function around plot_ly() with fitted(lm(y~x)) using add_lines()

r

function

lm

plotly

broom