library(tidyverse)
library(forcats)
我有两个简单的数据框（底部代码），我想通过折叠“动物”列创建一个新的已编码变量。我通常使用forcats :: fct_collapse进行此操作。但是，我想使一个函数将fct_collapse应用于具有相同变量的许多不同数据帧，除了一些可能缺少一个或两个因子水平。例如，在这种情况下，Df2缺少“犀牛”。
有什么办法可以更改代码（使用tiyverse），以便将缺少的因子类别返回为NA？在此示例中，我知道它是“犀牛”，但是在我的实际数据中可能还有其他缺失水平。除forcats :: fct_collapse外，我还接受其他选择，但我希望保留在tidyverse领域内。
REC <- function(Df, Data){
Df %>%
mutate(NEW = fct_collapse(Data, One = c("Cat","Dog","Snake"),
Two = c("Elephant","Bird","Rhino")))
}
REC(Df1,Animal) - this works
REC(DF2,Animal) - this doesn't, it throws an error because of "Rhino"
样本数据：
Animal <- c("Cat","Dog","Snake","Elephant","Bird","Rhino")
Code <- c(101,222,434,545,444,665)
Animal2 <- c("Cat","Dog","Snake","Elephant","Bird")
Code2 <- c(101,222,434,545,444)
Df1 <- data_frame(Code, Animal)
Df2 <- data_frame(Code2, Animal2) %> %rename(Animal = Animal2)
这是给你的一个主意。最初，我尝试在函数中包含两个参数。一个用于数据框，另一个用于包含动物名称的列。但是这次尝试失败了。我收到一条错误消息，指出“ mutate_impl（.data，点）中的错误：列的new长度必须为5（行数）或一个，而不是6。” 因此，我决定在函数中不使用列名。我Animal在职能上明确表示。然后，一切正常。这个想法是创建一个缺少动物名字的因子变量。那是在做factor()用setdiff()。一旦有了所有动物的名字，我就用fct_collapse()。
myfun <- function(mydf){
animals <- c("Cat", "Dog", "Snake", "Elephant", "Bird", "Rhino")
mydf %>%
mutate(new = factor(Animal, levels = c(unique(Animal), setdiff(animals, Animal))),
new = fct_collapse(new, One = c("Cat", "Dog", "Snake"),
Two = c("Elephant", "Bird", "Rhino"))) -> x
x}
> myfun(Df2)
# A tibble: 5 x 3
Code2 Animal new
<dbl> <chr> <fct>
1 101 Cat One
2 222 Dog One
3 434 Snake One
4 545 Elephant Two
5 444 Bird Two
> myfun(Df1)
# A tibble: 6 x 3
Code Animal new
<dbl> <chr> <fct>
1 101 Cat One
2 222 Dog One
3 434 Snake One
4 545 Elephant Two
5 444 Bird Two
6 665 Rhino Two
备注：除了我有两个参数外，以下函数是相同的。这是行不通的。如果有任何修改的可能，请告诉我。
myfun2 <- function(mydf, mycol){
animals <- c("Cat", "Dog", "Snake", "Elephant", "Bird", "Rhino")
mydf %>%
mutate(new = factor(mycol, levels = c(unique(mycol), setdiff(animals, mycol))),
new = fct_collapse(new, One = c("Cat", "Dog", "Snake"),
Two = c("Elephant", "Bird", "Rhino"))) -> x
x}
> myfun2(Df2, Animal)
Error in mutate_impl(.data, dots) :
Column `new` must be length 5 (the number of rows) or one, not 6
本文收集自互联网，转载请注明来源。
如有侵权，请联系 [email protected] 删除。
我来说两句