Início / coding / Perguntas / 78856743
Accepted
farrow90
Asked: 2024-08-11 02:17:38 +0800 CST

Aplicando funções em uma linha

  • 772

Eu tenho este conjunto de dados:

set.seed(123)  
old <- data.frame(matrix(sample(c(1:10, NA), 35, replace = TRUE), ncol = 7, nrow = 5))

Em cada linha, quero encontrar os primeiros 5 valores não-NA e colocá-los em um novo dataframe.

Pensei em como fazer isso com a seguinte função:

extract_non_na <- function(row) {
    non_na_values <- na.omit(row)
    if (length(non_na_values) < 5) {
        non_na_values <- c(non_na_values, rep(NA, 5 - length(non_na_values)))
    }
    return(non_na_values[1:5])
}

O problema agora era como aplicar essa função nas linhas de R. Eu investiguei isso e aparentemente isso é feito com t?

new <- t(apply(old, 1, extract_non_na))

Este é o protocolo correto?

1 respostas

  1. Best Answer

    Se você quiser ter ações em linhas, você pode usar applycomo abaixo

    > data.frame(t(apply(old, 1, \(x) x[order(is.na(x))][1:5])))
  X1 X2 X3 X4 X5
1  3 10  9 10  4
2  3  5  9  7  1
3 10  4  5  9 10
4  2  6  3  3  9
5  6  9  8  3  5

    Ou você pode tentar order+ rowcomo abaixo

    > data.frame(matrix(unlist(old)[order(row(old), is.na(old))], nrow = nrow(old), byrow = TRUE))[1:5]
  X1 X2 X3 X4 X5
1  3 10  9 10  4
2  3  5  9  7  1
3 10  4  5  9 10
4  2  6  3  3  9
5  6  9  8  3  5
    • 1

relate perguntas