multifarm/intro_r

Fork 0

This file contains Unicode characters that might be confused with other characters. If you think that this is intentional, you can safely ignore this warning. Use the Escape button to reveal them.

Show All Code
Hide All Code

Criar uma coluna com caracteres indicando qual célula em três colunas da tabela apresenta maior valor

José

2021-11-13

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 é usado 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)
           ))

</html>