Категоризируйте несколько строк в одну переменную

Question

Простой вопрос, но, по-видимому, еще не ответил в StO.

У меня длинный кадр данных, где 3 из колонок:

person | trip | driver 
======================= 
1  car 
1  bike 
1  train 
1  walk 
2  walk 
2  train 
2  boat 

То, что я хотел бы, чтобы заполнить «драйвер» столбца, так что он читает 1, если по крайней мере один из поездок производится на машине, 0 в противном случае:

person | driver 
================ 
1  1 
1  1 
1  1 
1  1 
2  0 
2  0 
2  0 

у меня есть небольшое предпочтение делать это, не прибегая к фантазии пакетов, но я счастлив с большинством популярных (например, plyr, data.table, sqldf ....) или даже новые, которые оказываются полезными в долгосрочной перспективе ,

Заранее благодарю .p.

Answer 1

Мы могли бы использовать data.table, конвертировать «data.frame» до «data.table» (setDT(df1)), мы проверяем, есть ли any «автомобиль» в «путешествие» сгруппированных по «человека», преобразовать логический вывод (+0L или обертывание с помощью as.numeric) и назначить (:=) колонке «драйвер». При необходимости, мы можем удалить столбец «отключения», назначив ее NULL или подмножества по [, c(1,3), with=FALSE]

library(data.table) 
setDT(df1)[, driver := any(trip == 'car')+0L, by = person][, trip := NULL] 

Или вместо any, мы можем использовать в качестве max(trip=='car') @Arun упоминается в комментариях

setDT(df1)[, driver := max(trip == 'car'), by = person] 

---

Или используя аналогичную логику, как описано выше, мы создаем новую колонку с mutate и удаляем ненужные столбцы с select

library(dplyr) 
df1 %>% 
    group_by(person) %>% 
    mutate(driver= any(trip=='car')+0L) %>% 
    select(-trip) 

---

Или с base R, мы можем использовать ave создать «драйвер», а затем subset, чтобы удалить столбец «отключение».

df1$driver <- with(df1, ave(trip=='car', person, FUN=any)+0L) 
subset(df1, select=-trip)