r - 在 apply 内使用 ifelse

Question

我试图在我的数据集中创建一个新列，为每一行提供一个输出，具体取决于来自预先存在的列的输入。

在这个输出列中，我想要“NA”如果 给定行中的任何输入值都是“0”。 否则 (如果所有输入都不为 0)，我希望该行的输出是输入的唯一值的数量。

我认为该解决方案将使用 ifelse 嵌套在 中的函数申请 函数，但我收到一个我不明白的错误。

data$output <- apply(data, 1, function(x) {ifelse(x == 0, NA, length(unique(x)))})

Error in $<-.data.frame(*tmp*, "output", value = c(3L, 3L, 3L, 3L, : replacement has 3 rows, data has 4

我不知道为什么替换有 3 行，因为我认为 apply 只是对我的 4 行中的每一行执行相同的功能。

Answer 1

您想检查一行中是否有任何变量为 0，因此您需要使用 any(x==0)而不是 x == 0在 ifelse陈述:

apply(data, 1, function(x) {ifelse(any(x == 0), NA, length(unique(x)))})
# [1]  1 NA  2

基本上 ifelse如果第一个参数的长度为 n，则返回长度为 n 的向量。您需要每行一个值，但使用 x==0 传递了多个值(您传递的值数等于数据框中的列数)。

数据:

(data <- data.frame(a=c(1, 2, 3), b=c(1, 0, 1)))
#   a b
# 1 1 1
# 2 2 0
# 3 3 1