r - 基于数据框中的两个旧列创建一个新列

Question

data <- data.frame(foo = c(0, 1, 0, 0, 1, 0, 0, 0, 1, 1, 1),
                   bar = c(1, 0, 0, 0, 1, 0, 1, 0, 0, 1, 0))

您好，我这里有一个包含两列 foo 和 bar 的数据框。我想根据 foo 和 bar 数据创建一个新的 Complete 列。

• 如果 foo 和 bar 为零，则 complete 应为 0。
• 如果 foo 是 1 而 bar 是 0 那么 complete 应该是 1。
• 如果 bar 是 1 且 foo 是 0 那么 complete 应该是两个。

例如。

foo   bar complete
0     0   0
1     0   1
0     1   2

编辑:

如果 foo==1 和 bar==1 则 NA。

Answer 1

依此类推，当两列均为 1 时使用 NA。从行总和开始。如果其中任何一个为 2(列数)，则将其替换为 NA。然后将其乘以 max.col() 值。

rs <- rowSums(data)
cbind(data, complete = max.col(data) * replace(rs, rs == 2, NA))
#    foo bar complete
# 1    0   1        2
# 2    1   0        1
# 3    0   0        0
# 4    0   0        0
# 5    1   1       NA
# 6    0   0        0
# 7    0   1        2
# 8    0   0        0
# 9    1   0        1
# 10   1   1       NA
# 11   1   0        1

如果您不想分配新对象，您可以使用本地环境或将其包装到一个函数中:

local({
    rs <- rowSums(data)
    max.col(data) * replace(rs, rs == 2, NA)
})
# [1]  2  1  0  0 NA  0  2  0  1 NA  1