Выберите имя столбца первого столбца, который соответствует определенный логический тест

Question

я следующий вход:

id <- c("a", "b", "c", "d") 
target <- seq(from = 100, to = 400, length.out = 4) 
a <- c(300, 304, 100, 405) 
b <- c(300, 104, 100, 405) 
c <- c(85, 304, 500, 405) 
df <- as.data.frame(cbind(id, target, a, b, c)) 

Я хотел бы добавить новый столбец «столбец», который указывает на строку, которая из столбцов «а» , «b», «c» будет первым столбцом со значением, меньшим целевого решения. Запрошенный результат выглядит следующим образом:

Обязательный выход:

df$column <- c("c", "b", "a", "NA") 
df 

Я подумал про concenated если чек на строку и применить это ко всем строкам с помощью функции применяются. Однако столбцы abc довольно длинные (круглый 20, поэтому потребуется цикл), а число строк - около 4.000. Есть ли у кого-нибудь идея о том, как его решить?

Answer 1

Вот еще Векторизованное решения с использованием which. Это в основном принимает все случаи, когда target больше, а первые экземпляры используются с помощью функции duplicated.

indx <- which(df[, 3:5] < df[, 2], arr.ind = TRUE) 
indx2 <- indx[!duplicated(indx[, "row"]),] 
df[indx2[, "row"], "column"] <- names(df)[3:5][indx2[, "col"]] 
df 
# id target a b c column 
# 1 a 100 300 300 85  c 
# 2 b 200 304 104 304  b 
# 3 c 300 100 100 500  a 
# 4 d 400 405 405 405 <NA> 

Answer 2

Вы можете сделать это следующим образом:

1) Создать логическую матрицу, указывающую или нет значение в «A», «B» или «с» столбец меньше, чем целевой столбец:

m <- df[,3:5] < df[,2] 

2) Создать целочисленный вектор, который является первыми именами этих трех столбец, который имеет значение, меньший, чем целевой столбец с max.col и убедитесь, что значение NA возвращается для строк, где не было никакого значения меньше с [c(TRUE,NA)[1 + (rowSums(m) == 0)]]:

mc <- max.col(m, ties.method = 'first')[c(TRUE,NA)[1 + (rowSums(m) == 0)]] 

3) Присвоить имена в новый столбец:

df$column <- names(df[,3:5])[mc] 

, который дает:

> df 
    id target a b c column 
1 a 100 300 300 85  c 
2 b 200 304 104 304  b 
3 c 300 100 100 500  a 
4 d 400 405 405 405 <NA> 

---

Я отделил шаги, чтобы сделать его более ясным что делает код. Но вы можете, конечно, интегрировать его более следующим образом:

m <- df[,3:5] < df[,2] 
df$column <- names(df[,3:5])[max.col(m, ties.method = 'first')[c(TRUE,NA)[1 + (rowSums(m) == 0)]]] 

Answer 3

Вы можете применить функцию вдоль строк, например. и использовать результат для заполнения вашей колонки т.д.,

searchFunction <- function(row) { 
    result <- "NA" 
    for (name in names(row)) { 
    if (name == "target" || name == "id") { 
     next 
    } 
    if (result == "NA" && as.numeric(row[name]) < as.numeric(row["target"])) { 
     result = name 
    } 
    } 
    return(result); 
} 

apply(df, 1, searchFunction) 
# [1] "c" "b" "a" "NA"