Применить функцию match и replace над сериями строк в фрейме данных по порядку

запуск фрейма данных

data_start <- data.frame(marker = c("yes","yes","no","yes","no"),
                         id_out = c(5,3,1,1,7), 
                         id_new = c(6,8,9,4,2))

> data_start
  marker id_out id_new
1    yes      5      6
2    yes      3      8
3     no      1      9
4    yes      1      4
5     no      7      2

добавить три заголовка столбцов с пустыми столбцами ниже. Прикрепите начало var1:var3 значения.

data_start[,c("var1", "var2", "var3")] <- NA
vars <- c(5,3,1)
data_start[1, 4:6] <- vars

> data_start
  marker id_out id_new var1 var2 var3
1    yes      5      6    5    3    1
2    yes      3      8   NA   NA   NA
3     no      1      9   NA   NA   NA
4    yes      1      4   NA   NA   NA
5     no      7      2   NA   NA   NA

я хотел бы обновить мой var1:var3 столбцы, применяя функцию к каждой строке, где IF marker = yes и id_out совпадает с var1:var3, заменить на var1:var3 С id_new. Я нашел это решение, но работает для одной строки кода и по-прежнему требует от каждого нового var1:var3 часть строки к обновление.

data_start[1, 4:6][data_start[1, 4:6] == data_start[1,"id_out"]] <- data_start[1,"id_new"]

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

конечный результат будет выглядеть так, где строки остаются неизменными, когда маркер = no и каждая строка впоследствии обновляется.

> data_final
  marker id_out id_new var1 var2 var3
1    yes      5      6    6    3    1
2    yes      3      8    6    8    1
3     no      1      9    6    8    1
4    yes      1      4    6    8    4
5     no      7      2    6    8    4
10
apply dataframe r

4 ответов

Это можно использовать с любым количеством столбцов и работает с базой Р:

cols <- c("var1", "var2", "var3")

for(j in 1:length(cols)) {
  var <- cols[j]
  for(i in 1:nrow(data_start)){
    if(i > 1) {
      data_start[i, var] <- data_start[i-1, var]
    }
    if(data_start[i, "marker"] == "yes" & data_start[i, var] == data_start[i,"id_out"]) {
      data_start[i,var] <- data_start[i, "id_new"]
    } 
  }
}
1

это очень грубо, потому что мне нужно бежать, но это должно сработать.

data_start <- data.frame(marker = c("yes","yes","no","yes","no"),
                         id_out = c(5,3,1,1,7), 
                         id_new = c(6,8,9,4,2))

data_start[,c("var1", "var2", "var3")] <- NA
vars <- c(5,3,1)
data_start[1, 4:6] <- vars

onVars <- c("var1", "var2", "var3")

for (i in 2:nrow(data_start)) {

  print(i)

  for (var in onVars) {

    if (data_start$marker[i] == "yes" & data_start$id_out[i] == data_start[i - 1, var]) {

      data_start[i, var] <- data_start$id_new[i]

    } else {

      data_start[i, var] <- data_start[i - 1, var]

    }

  }

}

data_start-это ваш выход.

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

0

кажется трудным найти решение без цикла for, и если, то вот оно. Я попытался изменить исходные значения с помощью другого набора, такого как c(1,3,1) и коды работают нормально. При необходимости можно также добавить дополнительные столбцы переменных.

# Re-create the data
dt <- data.table(marker = c("yes","yes","no","yes","no"),
                 id_out = c(5,3,1,1,7),
                 id_new = c(6,8,9,4,2))
var.col <- paste0("var", 1:3)
dt[1, (var.col) := .(5,3,1)]

# Processing
for(i in 1:nrow(dt)) {
  if(i > 1) dt[i, (var.col) := as.list(dt[i-1, var.col, with = F])]
  var.i <- dt[i, var.col, with = F] %in% dt[i, id_out]
  if(dt[i]$marker == 'yes' & sum(var.i) != 0) {
    dt[i, (var.col[var.i]) := dt[i, id_new]]
  }
}
0

вот фрагмент, который позволит вам сделать этот расчет, даже если у вас более трех столбцов:

library(data.table)
dt <- data.table(marker = c("yes","yes","no","yes","no"),
                         id_out = c(5,3,1,1,7), 
                         id_new = c(6,8,9,4,2))

dt[, change := cumsum(marker == "yes")]

ref.new <- dt[marker == "yes", id_new] # Reference to values where marker is "yes"
ref.out <- dt[marker == "yes", id_out]
for (x in 1:length(ref.new)) {
  dt[, paste("var", x, sep="") := ifelse(change >= x, ref.new[x] , ref.out[x])]
}
head(dt)
#     marker id_out id_new change var1 var2 var3
#1:    yes      5      6      1    6    3    1
#2:    yes      3      8      2    6    8    1
#3:     no      1      9      2    6    8    1
#4:    yes      1      4      3    6    8    4
#5:     no      7      2      3    6    8    4
0