Блог

Главная - Вопросы по программированию - преобразование одного столбца фактора в несколько дихотомических столбцов в r

преобразование одного столбца фактора в несколько дихотомических столбцов в r

#r #dataframe #dplyr #converters

#r #фрейм данных #dplyr #конвертеры

Вопрос:

У меня есть набор данных с идентификатором пациента и их диагнозами, и они следующие :

 Id  Diagnoses
1   Nerve conditions (e.g., Multiple sclerosis, myasthenia gravis, Guillain-Barre syndrome, demyelinating polyneuropathy)
2   Gastrointestinal conditions (e.g., irritable bowl disease, ulcerative colitis, Chron's disease),Heart conditions,High blood pressure,Migraines/headaches
3   Heart conditions,Traumatic brain injury
4   Chronic pain,Heart conditions,Post-traumatic Stress Disorder (PTSD),Traumatic brain injury
5   Anxiety,Chronic pain,Depression,Sleep apnea
6   High blood pressure
7   High blood pressure

Как я могу разделить Diagnoses столбец следующим образом :

 Id  Anxiety       Depression      Nerve conditions  Sleep apnea Chronic Diseases  AND SO ON....
1   0               0              0                 1            1                
2   1               1              1                 1            1             
3   1               1              1                 1            0
4   0               0              1                 1            1          
5   1               0              0                 0            1         
6   1               1              1                 0            1         
7   1               1              0                 1            0

Я пробовал этот код, но я не получил результата:

 df %>%
   separate_rows(Diagnoses, sep=",") %>% 
   separate(Q2.3, into = c("Anxiety", "Depression, "THE REST OF CONDITIONS"), sep=":s*") %>% 
   mutate(anxiety1 = str_c("Anxiety", Anxiety))

Я ценю вашу помощь.,

Комментарии:

1. Как это High blood pressure дает 1 для Anxiety и Depression ? Почему строки 6 и 7 имеют 1 в разных местах, когда их значение в Diagnoses одинаковое.

2. Извините за путаницу. Это был пример, и он не был точным. Я просто хочу показать вам, ребята. Еще раз извините.

Ответ №1:

Работает ли это:

 library(stringr)
library(dplyr)
library(tidyr)
df %>% mutate(Diagnoses = str_remove(Diagnoses, ' \(.*\)?')) %>% 
separate_rows(Diagnoses, sep = ',') %>% count(Id, Diagnoses, name = 'Cnt') %>% 
pivot_wider(id_cols = Id, names_from = Diagnoses, values_from = Cnt, values_fill = list(Cnt = 0))
# A tibble: 7 x 11
     Id `Nerve condition~ `Gastrointestina~ `Heart conditio~ `Traumatic brai~ `Chronic pain` `Post-traumatic ~ Anxiety Depression `Sleep apnea` `High blood pre~
  <dbl>             <int>             <int>            <int>            <int>          <int>             <int>   <int>      <int>         <int>            <int>
1     1                 1                 0                0                0              0                 0       0          0             0                0
2     2                 0                 1                0                0              0                 0       0          0             0                0
3     3                 0                 0                1                1              0                 0       0          0             0                0
4     4                 0                 0                1                0              1                 1       0          0             0                0
5     5                 0                 0                0                0              1                 0       1          1             1                0
6     6                 0                 0                0                0              0                 0       0          0             0                1
7     7                 0                 0                0                0              0                 0       0          0             0                1
>