Как использовать apply для изменения элементов одного информационного кадра, основанного на столбцах другого? - PullRequest
0 голосов
/ 27 февраля 2020

У меня есть фрейм данных, где два столбца отмечают начало и конец областей, которые мне нужно манипулировать в другом фрейме данных. Вместо применения 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 слишком велико, недостаточно памяти для выделения матрицы, полученной в результате применения.

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

Спасибо!

Ответы [ 2 ]

2 голосов
/ 27 февраля 2020

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

logint <- rep(FALSE,50)
logint[unlist(Map(`:`, df$b, df$e))] <- TRUE
0 голосов
/ 27 февраля 2020

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

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