如何使我的for循环仅返回ifelse语句的第一个匹配项?
这是我的数据框:
df <- structure(
list(
x0 = c("0", "0", "0", "0", "0", "0", "0", "0",
"0", "0"),
x1 = c("0", "0", "0", "0", "0", "0", "0", "0", "0",
"0"),
x2 = c("0", "0", "0", "0", "0", "0", "0", "0", "0", "0"),
x3 = c("0", "0", "0", "0", "1", "0", "0", "0", "0", "0"),
x4 = c("0", "0", "0", "0", NA, "0", "0", "0", NA, "0"),
x5 = c("0",
"0", "0", "0", NA, "0", "0", "0", NA, "0"),
x6 = c("0", "0",
"0", "0", NA, "0", "0", "0", NA, "0"),
x7 = c("0", "0", "0",
"0", NA, "0", "0", "0", NA, "0"),
x8 = c("0", "0", "0", "0",
NA, "0", "0", "0", NA, "0"),
x9 = c("0", "0", "0", "0", NA,
"0", "0", "0", NA, "1"),
x10 = c("0", "0", "0", "0", NA,
"0", "0", "1", NA, "1"),
x11 = c("0", "0", "0", "0", NA,
"0", "0", "1", NA, "2"),
x12 = c("0", "0", "0", "0", NA,
"0", "0", "0", NA, "2"),
x13 = c("0", "0", "0", "0", NA,
"0", "0", "1", NA, "3"),
x14 = c(NA, NA, "0", "0", NA, NA,
NA, "2", NA, NA),
x15 = c(
NA_character_,
NA_character_,
NA_character_,
NA_character_,
NA_character_,
NA_character_,
NA_character_,
NA_character_,
NA_character_,
NA_character_
)
),
.Names = c(
"x0",
"x1",
"x2",
"x3",
"x4",
"x5",
"x6",
"x7",
"x8",
"x9",
"x10",
"x11",
"x12",
"x13",
"x14",
"x15"
),
row.names = c(NA, 10L),
class = "data.frame"
)
这是我想要实现的预期输出:
df$expectedoutput <- c(NA, NA, NA, NA, NA, NA, NA, 'x14', NA, 'x11')
这是我尝试获得上述输出的尝试:
df$newvar <- NA
for (i in 1:16) {
df$newvar <- ifelse(df[,i] == "2" & is.na(df$newvar),
colnames(df[i]), NA)
}
我相信,我的问题是,for循环,我已经创建的推移到底,覆盖发生的第一场比赛,例如用于排10,而不是在列名完成了df[10,12]
循环继续df[10,16]
,因此返回NA
但是,我可能错了,也许这不是我的问题。
首先,使用apply
获取第一个2的列索引。
其次,获取与该索引对应的名称
first2 <- apply(df, MARGIN = 1, FUN = function(x) head(which(x == 2), 1))
df$newvar <-
vapply(first2,
FUN = function(x) if (length(x)) names(df)[x] else NA_character_,
character(1))
该for
循环相当于将
df$newvar <- NA
for (i in seq_len(nrow(df))){
first2 <- head(which(df[i, ] == 2), 1)
df$newvar[i] <- if (length(first2)) names(df)[first2] else NA
}
本文收集自互联网,转载请注明来源。
如有侵权,请联系 [email protected] 删除。
我来说两句