我正在尝试研究如何创建用户定义的函数来对数据框中的一系列列执行计算，并将答案作为附加列添加到同一数据框中。为简单起见，我一直使用的测试示例是计算从一年到下一年的百分比增长，但目标是能够创建更复杂的计算，这些计算过于繁琐和重复而无法手动计算。
我一直在使用的练习数据是...
a <- c(10, 12)
b <- c(11, 9)
df <- t(data.frame(a, b))
df <- data.frame(df)
colnames(df) <- c(2001, 2002))
哪个看起来像...
2001 2002
a 10 12
b 11 9
我一直在使用的手动计算是...
df$PercGrowth <- (df$`2002` - df$`2001`) / df$`2001` * 100
返回：
2001 2002 PercGrowth
a 10 12 20.00000
b 11 9 -18.18182
如何将其转换为用户定义的函数，我可以在其中指定要执行计算的列，然后将答案作为派生值添加到数据框中？
我最初认为可能有效的是......
pg <- function(data, c1, c2)
df <- mutate(data, PercGrowth = ((df[c2] -df[c1]) / df[c1] * 100))
pg(df, 1, 2)
但是我不断收到错误消息：
错误：列PercGrowth属于不受支持的类 data.frame
我如何让这个工作？
这实际上比看起来更复杂 - 您需要使用 dplyr 代词和 quasiquotation 以便将列名作为函数中的参数传递。以下代码有效：
library(dplyr)
a <- c(10, 12)
b <- c(11, 9)
df <- t(data.frame(a, b))
df <- data.frame(df)
colnames(df) <- c("year1", "year2")
pg <- function(df, col1, col2) {
quo_col1 <- enquo(col1)
quo_col2 <- enquo(col2)
df %>%
mutate(pct_growth = (!! quo_col2 - !! quo_col1) / !! quo_col1 * 100)
}
pg(df, year1, year2)
我将列重命名为字符串，以便它们更易于使用。您可以在此链接中阅读更多内容：https : //dplyr.tidyverse.org/articles/programming.html
另一种选择是在您感兴趣的列名称上使用某种字符串匹配，使用这些列执行操作，然后将结果连接回主数据框。
本文收集自互联网，转载请注明来源。
如有侵权，请联系 [email protected] 删除。
我来说两句