Объединение наборов столбцов фрейма данных - PullRequest
Новая
       7

Объединение наборов столбцов фрейма данных

0 голосов
/ 26 января 2019

Возможно, простой.

У меня есть такой тип data.frame: 

df <- data.frame(sp1.name = c("sp1.n1",NA,"sp1.n3",NA), sp1.id = c("sp1.id1","sp1.id2",NA,NA),
                 sp2.name = c(NA,NA,"sp2.n3",NA), sp2.id = c(NA,NA,NA,"sp2.id4"),
                 sp3.name = c("sp3.n1",NA,NA,NA), sp3.id = c("sp3.id1",NA,NA,NA))

Он состоит из пар столбцов для каждого индекса 'sp': sp<index>.nameи sp<index>.id.В этом примере индекс равен 1,2,3.

Я ищу способ (возможно, через tidyverse) объединить для каждой sp свою пару соответствующих столбцов name и id, гдеПравила слияния:

  1. if !is.na(sp<index>.name) & !is.na(sp<index>.id) return sp<index>.name
  2. if !is.na(sp<index>.name) & is.na(sp<index>.id) return sp<index>.name
  3. else if is.na(sp<index>.name) & !is.na(sp<index>.id) return sp<index>.id
  4. else return NA

Итак, для этого примера результирующее data.frame: 

df <- data.frame(sp1 = c("sp1.n1","sp1.id2","sp1.n3",NA),
                 sp2 = c(NA,NA,"sp2.n3","sp2.id4"),
                 sp3 = c("sp3.n1",NA,NA,NA))

1 Ответ

0 голосов
/ 27 января 2019

Вы можете сделать: 

library(tidyverse)

df %>%
  mutate(rn = row_number()) %>%
  gather(id, value, -rn) %>%
  mutate(idx = gsub("\\..*", "", id)) %>%
  group_by(idx, rn) %>%
  mutate(
    value = case_when(
      any(grepl("name", id) & !is.na(value)) & any( (grepl("id", id) & !is.na(value)) | (grepl("id", id) & is.na(value)) ) ~ value[grepl("name", id)],
      any(grepl("name", id) & is.na(value)) & any(grepl("id", id) & !is.na(value)) ~ value[grepl("id", id)],
      TRUE ~ NA_character_)) %>% 
  distinct(idx, value, rn) %>%
  spread(idx, value)

Предоставление: 

# A tibble: 4 x 4
# Groups:   rn [4]
     rn sp1     sp2     sp3   
  <int> <chr>   <chr>   <chr> 
1     1 sp1.n1  NA      sp3.n1
2     2 sp1.id2 NA      NA    
3     3 sp1.n3  sp2.n3  NA    
4     4 NA      sp2.id4 NA
...