Question

我需要计算何时在0和1之间切换的值，这些值跨列分布，没有给出切换，并且存在NA。

我尝试使用mutate和rowSums，但收效甚微。

示例：

df <- data.frame(entry = c(1:5), 
                year_1 = c(NA, NA, NA, 1, NA),
                year_2 = c(NA, NA, 0, 0, 1),
                year_3 = c(NA, 1, 1, 0, 1))

所需结果：

switch = c(NA, NA, "year_2", NA, NA)

Answer 1

l <- apply(df[, -1], 1, function(x) 
        names(df)[1 + which(tail(x, -1) == 1 & head(x, -1) == 0)])
unlist(ifelse(lengths(l), l, NA))

# [1] NA       NA       "year_2" NA       NA

Answer 2

要计算跨列的更改，可以采用数据的“超前”和“滞后”版本（列方式）之差。获取差异为1的索引，并使用这些索引创建“ switch”：

ix <- which(df[ , 3:ncol(df)] - df[ , 2:(ncol(df) - 1)] == 1, arr.ind = TRUE) 
df$switch <- NA
df$switch[ix[ , 1]] <- paste0("year_", ix[ , 2])

df
#   entry year_1 year_2 year_3 switch
# 1     1     NA     NA     NA   <NA>
# 2     2     NA     NA      1   <NA>
# 3     3     NA      0      1 year_2
# 4     4      1      0      0   <NA>
# 5     5     NA      1      1   <NA>

跨列的条件值更改

2 个答案: