Build a column with index of greater values per line

script/replace_values.R

@@ -0,0 +1,105 @@
+#' ---
+#' title: "Criar uma coluna com caracteres indicando qual célula em três colunas da tabela apresenta maior valor"
+#' date: "2021-11-13"
+#' author: "José"
+#' output:
+#'   html_document:
+#'    code_folding: show
+#'    toc: yes
+#'    toc_float:
+#'      smooth_scroll: true
+#'    df_print: paged
+#'    highlight: zenburn
+#' ---
+
+#' Remover objetos
+
+rm(list = ls())
+
+
+#' ## Pergunta
+#'
+#' Pergunta do grupo R-brasil no Telegram
+#'
+#' Quero criar a coluna "maior" com o nome da coluna que tem o maior
+#' valor na linha
+#'
+#' ## Solução usando R base
+#'
+#' ###  Criar uma tabela com três colunas de tipo numérico com valores  entre 0 a 1
+#'
+#' A função *runif* cria 10 valores entre 0 a 1 se os valores máximo e mínimo não são definidos dentro do parêntese
+#'
+df <- data.frame(a = runif(10),
+                 b = runif(10),
+                 c = runif(10))
+
+
+#' ### Criar vectores definindo a condição lógica esperada em cada caso
+
+#' Se a coluna "a" apresentar o maior valor, então o vector "a_maior"
+#' terá um elemento "TRUE", do contrário, o elemento será "FALSE"...
+
+a_maior <- with(df, a > b & a > c)
+
+b_maior <- with(df, b > a & b > c)
+
+c_maior <- with(df, c > a & c > b)
+
+#' ### Criar uma coluna na tabela com valores de cero em todas as linhas
+
+df$maior <- 0
+
+#' ### Indexar um caractere dentro da coluna "maior" indicando a coluna que apresentou o maior valor em cada linha da tabela
+#'
+#' Inserir "a" na posição em que a coluna "a" apresentou o maior valor; ou seja: quando a condição lógica foi igual a "TRUE"
+
+df$maior[a_maior] <- "a"
+
+df$maior[b_maior] <- "b"
+
+df$maior[c_maior] <- "c"
+
+#' ### Criar uma função simples para executar a tarefa
+
+maior_valor <- function (df, a, b, c, ...) {
+
+    a_maior <- with(df, a > b & a > c)
+    b_maior <- with(df, b > a & b > c)
+    c_maior <- with(df, c > a & c > b)
+
+    linhas <- nrow(df)
+    coluna <- rep(0, times = linhas)
+
+    coluna[a_maior] <- "a"
+    coluna[b_maior] <- "b"
+    coluna[c_maior] <- "c"
+
+    df$V1 <- coluna
+
+    return(df)
+}
+
+#' A coluna "maior" é o resultado de indexar com a primeira parte do' código e a coluna "V1" resulta de aplicar a função na tabela
+#'
+maior_valor(df, a, b, c)
+
+#' ## Usando o pacote dplyr
+
+library(dplyr)
+
+dfd <- tibble(p_col = runif(10),
+              s_col = runif(10),
+              t_col = runif(10))
+
+#' A função *rowwise* aplica a função *which.max* em todas as linhas.  *case_when* é usada para substituir o número da coluna pelo nome
+
+dfd %>%
+    rowwise() %>%
+    mutate(m = which.max(c(p_col, s_col, t_col)),
+           m = case_when(
+               m == 1 ~ names(.)[1],
+               m == 2 ~ names(.)[2],
+               m == 3 ~ names(.)[3],
+               TRUE ~ as.character(m)
+           ))