代码之家 › 专栏 › 技术社区 › Alexander

分组数据中的if语句

dplyr if-statement r

Alexander · 技术社区 · 8 年前

我正在尝试添加名为 'class' 我的数据基于列的条件。我构建了用户定义的函数 class_fun 以创建此新列。

如果条件匹配,我尝试将整个子组设置为相同的值。但我还没有成功!

test=data.frame(set=as.numeric(gl(3,6)),gr = rep(letters[1:2],each=3),vals=c(c(10,10, 9, 8, 1,1),c(10,10,10, 9,6,2),c(10,7,6,1,1,2)))

> test
   set gr vals
1    1  a   10
2    1  a   10
3    1  a    9
4    1  b    8
5    1  b    1
6    1  b    1
7    2  a   10
8    2  a   10
9    2  a   10
10   2  b    9
11   2  b    6
12   2  b    2
13   3  a   10
14   3  a    7
15   3  a    6
16   3  b    1
17   3  b    1
18   3  b    2

下面是创建名为 `class` 是的。

class_fun <- function(gr,set,vals){
  if(any(grepl("a|b",gr)&set==vals)){

    "catched"

  }
  else{

    NA
  }
}

library(dplyr)
test%>%
  group_by(set,gr)%>%
  mutate(class=class_fun(gr,set,vals))

它给予

# A tibble: 18 x 4
# Groups:   set, gr [6]
     set gr     vals class
   <dbl> <fct> <dbl> <lgl>
 1     1 a        10 NA   
 2     1 a        10 NA   
 3     1 a         9 NA   
 4     1 b         8 NA   
 5     1 b         1 NA   
 6     1 b         1 NA   
 7     2 a        10 NA   
 8     2 a        10 NA   
 9     2 a        10 NA   
10     2 b         9 NA   
11     2 b         6 NA   
12     2 b         2 NA   
13     3 a        10 NA   
14     3 a         7 NA   
15     3 a         6 NA   
16     3 b         1 NA   
17     3 b         1 NA   
18     3 b         2 NA

我所期待的是

# A tibble: 18 x 4
# Groups:   set, gr [6]
     set gr     vals class
   <dbl> <fct> <dbl> <lgl>
 1     1 a        10 NA   
 2     1 a        10 NA   
 3     1 a         9 NA   
 4     1 b         8 catched   
 5     1 b         1 catched
 6     1 b         1 catched
 7     2 a        10 NA   
 8     2 a        10 NA   
 9     2 a        10 NA   
10     2 b         9 catched   
11     2 b         6 catched   
12     2 b         2 catched  
13     3 a        10 NA   
14     3 a         7 NA   
15     3 a         6 NA   
16     3 b         1 NA   
17     3 b         1 NA   
18     3 b         2 NA

有什么想法或解决办法可以解释为什么会这样? 谢谢!

1 回复 | 直到 8 年前

BENY 8 年前

我认为不需要编写函数。

test%>%group_by(set,gr)%>%mutate(calss=ifelse(any(stringr::str_detect(gr,'a|b')&(set==vals)),'catched',NA)
+ )
# A tibble: 18 x 4
# Groups:   set, gr [6]
     set     gr  vals   calss
   <dbl> <fctr> <dbl>   <chr>
 1     1      a    10    <NA>
 2     1      a    10    <NA>
 3     1      a     9    <NA>
 4     1      b     8 catched
 5     1      b     1 catched
 6     1      b     1 catched
 7     2      a    10    <NA>
 8     2      a    10    <NA>
 9     2      a    10    <NA>
10     2      b     9 catched
11     2      b     6 catched
12     2      b     2 catched
13     3      a    10    <NA>
14     3      a     7    <NA>
15     3      a     6    <NA>
16     3      b     1    <NA>
17     3      b     1    <NA>
18     3      b     2    <NA>

分组数据中的if语句

下面是创建名为 class 是的。

下面是创建名为 `class` 是的。