我的数据如下所示:
library(tidyverse)
df <- tribble(
~y_val, ~z_val,
2, 4,
5, 3,
8, 2,
1, 1,
9, 3)
我有自定义函数fun_b()
,希望通过dplyr :: mutate调用将其应用于数据框。但是,fun_b()
使用fun_a()
内部有循环的函数:
fun_a <- function(x, y, z, times = 1) {
df <- data.frame()
for (i in 1:times) {
x <- x * 2 + i * x
y <- y / 3 + i * y
z <- z + 1 + z * i
d <- data.frame(x, y, z)
df <- rbind(df, d)
}
return(df)
}
fun_b <- function(x, y, z, times = 1) {
df <- fun_a(x, y, z, times)
x_r <- sum(df$x)
y_r <- sum(df$y)
z_r <- sum(df$z)
val <- x_r / y_r * z_r
return(val)
}
当我运行自定义函数时:
df %>%
mutate(test = fun_b(x = 1, y = y_val, z = z_val, times = 1))
中的每个突变值都test
显示相同的值(13.95)。那没有道理!例如,小标题的第一行(y_val = 2, z_val = 4)
应为10.125!
fun_b(x = 1, y = 2, z = 4, times = 1)
这里发生了什么?
您可以按行分组,以便针对每行分别对函数进行求值:
df %>%
rowwise() %>%
mutate(test = fun_b(x = 1, y = y_val, z = z_val, times = 1))
## Source: local data frame [5 x 3]
## Groups: <by row>
##
## # A tibble: 5 × 3
## y_val z_val test
## <dbl> <dbl> <dbl>
## 1 2 4 10.12500
## 2 5 3 3.15000
## 3 8 2 1.40625
## 4 1 1 6.75000
## 5 9 3 1.75000
或进行编辑fun_b
以使其向量化,或者只是让R:
df %>% mutate(test = Vectorize(fun_b)(x = 1, y = y_val, z = z_val, times = 1))
## # A tibble: 5 × 3
## y_val z_val test
## <dbl> <dbl> <dbl>
## 1 2 4 10.12500
## 2 5 3 3.15000
## 3 8 2 1.40625
## 4 1 1 6.75000
## 5 9 3 1.75000
本文收集自互联网,转载请注明来源。
如有侵权,请联系[email protected] 删除。
我来说两句