1

我想在 R 中围绕 plot_ly() 编写一个自定义函数。这样，​​我可以制作一系列具有相同格式和样式的散点图，但不能重复代码。我将此页面用作指南。此代码重现错误：

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"))) %>%

这是错误消息：

model.frame.default(formula = y ~ x, data = mtcars, drop.unused.levels = TRUE) 中的错误：对象不是矩阵

代码在不是函数时有效：

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")
4

1 回答 1

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)
于 2017-02-20T16:37:01.110 回答