Início / coding / Perguntas / 79339789
Accepted
JontroPothon
Asked: 2025-01-08 23:36:34 +0800 CST

Renomear múltiplas variáveis ​​com base em um vetor

  • 772

Eu tenho esse quadro de dados,

set.seed(123)
dataf <- data.frame( id = seq(1:5), 
                     A_1 = rnorm(5),
                     A_2 = rnorm(5),
                     A_3 = rnorm(5),
                     A_4 = rnorm(5),
                     A_5 = rnorm(5))

> dataf 
  id         A_1        A_2        A_3        A_4        A_5
1  1 -0.56047565  1.7150650  1.2240818  1.7869131 -1.0678237
2  2 -0.23017749  0.4609162  0.3598138  0.4978505 -0.2179749
3  3  1.55870831 -1.2650612  0.4007715 -1.9666172 -1.0260044
4  4  0.07050839 -0.6868529  0.1106827  0.7013559 -0.7288912
5  5  0.12928774 -0.4456620 -0.5558411 -0.4727914 -0.6250393

E eu tenho um vetor como o seguinte, com base no qual quero renomear as variáveis ​​do quadro de dados acima.

sep <- c(0, 1, 1, 0, 0)

Neste vetor, tenho o marcador com base no qual as variáveis ​​devem ser renomeadas. Se o marcador disser 0, nenhuma alteração. Se o marcador disser 1, adicionamos um prefixo ao nome da variável, digamos, _mod. Por exemplo, o segundo elemento de sepé 1, isso significa que preciso renomear a segunda variável A_2para A_2_mod. A tabela resultante deve ficar assim:

  id         A_1    A_2_mod    A_3_mod        A_4        A_5
1  1 -0.56047565  1.7150650  1.2240818  1.7869131 -1.0678237
2  2 -0.23017749  0.4609162  0.3598138  0.4978505 -0.2179749
3  3  1.55870831 -1.2650612  0.4007715 -1.9666172 -1.0260044
4  4  0.07050839 -0.6868529  0.1106827  0.7013559 -0.7288912
5  5  0.12928774 -0.4456620 -0.5558411 -0.4727914 -0.6250393

Posso fazer isso facilmente renomeando por posição. No entanto, estou pensando em como tornar a renomeação escalável no caso de, digamos, eu ter milhares de colunas.

3 respostas

  1. Best Answer

    Você pode usar dplyr::rename_withpara ajudar. Por exemplo

    dplyr::rename_with(dataf, 
   function(x) paste0(x, "_mod"), 
   .cols=num_range("A_", which(sep==1))
)

    Usei o num_range()auxiliar tidyselect para escolher apenas as colunas A que têm um 1 na posição correspondente emsep

    • 7

  2. Usando base::colnames:

    # include 1st column as zero - ie.don't change
x <- c(0, sep)
colnames(dataf)[ x == 1] <- paste0(colnames(dataf), "_mod")[ x == 1]

    Usando data.table::setnames:

    library(data.table)

setDT(dataf)
setnames(dataf,
         old = colnames(dataf)[ x == 1],
         new = paste0(colnames(dataf), "_mod")[ x == 1])
    • 5

  3. Tentar

    > names(dataf)[-1] <- paste0(names(dataf)[-1], c("", "_mod")[sep + 1])

> dataf
  id         A_1    A_2_mod    A_3_mod        A_4        A_5
1  1 -0.56047565  1.7150650  1.2240818  1.7869131 -1.0678237
2  2 -0.23017749  0.4609162  0.3598138  0.4978505 -0.2179749
3  3  1.55870831 -1.2650612  0.4007715 -1.9666172 -1.0260044
4  4  0.07050839 -0.6868529  0.1106827  0.7013559 -0.7288912
5  5  0.12928774 -0.4456620 -0.5558411 -0.4727914 -0.6250393
    • 5

relate perguntas