Как использовать apply для изменения элементов одного фрейма данных на основе столбцов другого?

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

df <- data.frame(b=c(7,25,32,44),e=c(11,27,39,48),n=c('a','b','c','d'))
logint <- rep(F,50)

log_vec <-  apply(df[,c('b','e')],1, function(x){logint[x['b']:x['e']] <- T;return(logint)})

Однако в результате получается матрица с одним столбцом для каждой строки df. Я знаю, что могу решить это с помощью

log_vec <- Reduce(`|`,as.data.frame(log_vec))

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

У вас есть лучшее решение?

Спасибо!


r apply
person RMelo    schedule 27.02.2020    source источник

Ответы (2)


arrow_upward
2
arrow_downward

Мы можем использовать mapply/Map для создания последовательности между значениями b и e и превратить их в TRUE.

logint <- rep(FALSE,50)
logint[unlist(Map(`:`, df$b, df$e))] <- TRUE
person Ronak Shah    schedule 27.02.2020

arrow_upward
0
arrow_downward

Мы также можем сделать это с помощью map2

library(dplyr)
library(purrr)
df %>% 
   transmute(new = map2(b, e, `:`)) %>%
   pull(new) %>% 
   flatten_int %>%
   replace(logint, ., TRUE)
person akrun    schedule 27.02.2020