Условный оператор if, основанный на значениях строк в r - PullRequest
Новая
       64

Условный оператор if, основанный на значениях строк в r

0 голосов
/ 03 февраля 2020

Я новичок в R, и я был бы очень признателен за вашу помощь в этом.

У меня есть фрейм данных, с двумя уровнями, которые обозначают 'Y' И 'N' для 11 переменных. enter image description here

Я хотел бы иметь новый столбец, который объединяет имена столбцов, когда значение строки равно 'Y'

т.е. enter image description here

Ответы [ 2 ]

4 голосов
/ 03 февраля 2020

Простой базовый R, способ сделать это: 

df1$newcol <- apply(df1, 1, function(x){
  paste(names(df1)[x == "Y"], collapse = "_")
})

Тестовый код создания данных. 

set.seed(1234)
df1 <- t(replicate(2, sample(c("N", "Y"), 10, TRUE)))
df1 <- as.data.frame(df1)
df1 <- cbind(df1, matrix(1:4, 2))
names(df1) <- paste0("col", 1:ncol(df1))
4 голосов
/ 03 февраля 2020

В базе R мы можем создать индексную матрицу строки / столбца, где значение равно "Y", используя which. Используя tapply, мы можем paste имена столбцов для каждой строки. 

cols <- paste0('col', 1:9)
mat <- which(df[cols] == 'Y', arr.ind = TRUE)
df$new_col <- as.character(tapply(names(df)[mat[, 2]], mat[, 1], 
               paste, collapse = "_"))
df
#  col1 col2 col3 col4 col5 col6 col7 col8 col9 col10 col11                       new_col
#1    N    Y    N    Y    Y    Y    N    Y    Y     1   624 col2_col4_col5_col6_col8_col9
#2    N    Y    N    Y    Y    Y    N    Y    N     7   548      col2_col4_col5_col6_col8

Используя tidyverse, мы можем получить данные в длинном формате, filter строк, где value равен "Y", и для каждого значения столбца вставки столбца. 

library(dplyr)

df %>%
  mutate(row = row_number()) %>%
  tidyr::pivot_longer(cols = -c(col10, col11, row)) %>%
  filter(value == 'Y') %>%
  group_by(row, col10, col11) %>%
  summarise(newcol = toString(name)) %>%
  ungroup() %>%
  select(-row)

данные 

df <- structure(list(col1 = structure(c(1L, 1L), .Label = "N", class = "factor"), 
col2 = structure(c(1L, 1L), .Label = "Y", class = "factor"), 
col3 = structure(c(1L, 1L), .Label = "N", class = "factor"), 
col4 = structure(c(1L, 1L), .Label = "Y", class = "factor"), 
col5 = structure(c(1L, 1L), .Label = "Y", class = "factor"), 
col6 = structure(c(1L, 1L), .Label = "Y", class = "factor"), 
col7 = structure(c(1L, 1L), .Label = "N", class = "factor"), 
col8 = structure(c(1L, 1L), .Label = "Y", class = "factor"), 
col9 = structure(2:1, .Label = c("N", "Y"), class = "factor"), 
col10 = c(1, 7), col11 = c(623.53, 548.028)), row.names = c(NA, -2L),
class = "data.frame")
...