Question

希望根据两列的值添加一列，但是多行。

示例数据集代码：

A = c(1,1,1,2,2,2,3,3,3,4,4)
B = c(1,2,3,1,2,3,1,2,3,1,2)
C = c(0,0,0,1,0,0,1,1,1,0,1)
data <- data.frame(A,B,C)

数据集：

Ifelse声明：

我想要实现的是“创建列D。如果列C = = 1 任何行，其中列A == x，列D = 1.其他列D == 0“

期望输出：

   A  B  C  D
1  1  1  0  0
2  1  2  0  0
3  1  3  0  0
4  2  1  1  1
5  2  2  0  1
6  2  3  0  1
7  3  1  1  1
8  3  2  1  1
9  3  3  1  1
10 4  1  0  1
11 4  2  1  1

我做了什么：

我今天已经考虑过这个问题，但无法提出合乎逻辑的答案，我已经尝试过查看长格式和宽格式的数据但是没有跳出来。

注意： 在实际应用中，x C列中显示的次数不相等（某些在数据集中包含一个重复，其他包含20个）。

Answer 1

# just check using any() if any group has a single row with C==1 

library(dplyr)
data %>% group_by(A) %>% mutate(D = as.numeric(any(C==1)))

library(data.table)
data[, D:=as.numeric(any(C==1)), by = .(A)]   
#       A     B     C     D
#1      1     1     0     0
#2      1     2     0     0
#3      1     3     0     0
#4      2     1     1     1
#5      2     2     0     1
#6      2     3     0     1
#7      3     1     1     1
#8      3     2     1     1
#9      3     3     1     1
#10     4     1     0     1
#11     4     2     1     1

Answer 2

轻松data.table

library(data.table)
data <- data.table(data)
x=2
data[,D:=ifelse(!A==x,ifelse(C==1,1,0),0)]
data

Answer 3

我们可以使用ave

中的base R

data$D <- with(data, as.integer(ave(C==1, A, FUN=any)))
data
#   A B C D
#1  1 1 0 0
#2  1 2 0 0
#3  1 3 0 0
#4  2 1 1 1
#5  2 2 0 1
#6  2 3 0 1
#7  3 1 1 1
#8  3 2 1 1
#9  3 3 1 1
#10 4 1 0 1
#11 4 2 1 1

跨多行的ifelse语句

3 个答案: