Somme des lignes dans les colonnes avec des noms de colonnes se terminant par une chaîne de caractères spécifique (R) [dupliquer]

Aug 18 2020

J'utilise le code suivant, qui calcule les scores z par sous-groupe. Je voudrais ajouter une colonne de plus à la fin, qui additionne les scores z, donc les sommes par ligne de toutes les valeurs dans toutes les colonnes qui se terminent par "_zscore". Comment pourrais-je sélectionner spécifiquement ces colonnes?

(Notez que mes données réelles ont beaucoup plus de colonnes, donc je cherche à sélectionner spécifiquement "_zscore" dans les noms de colonnes.)

 library(dplyr)
 set.seed(12345)
 df1 = data.frame(a=c(rep("a",8), rep("b",5), rep("c",7), rep("d",10)), 
      b=rnorm(30, 6, 2), 
      c=rnorm(30, 12, 3.5), 
      d=rnorm(30, 8, 3)
      )
 df1_z <- df1 %>%
   group_by(a) %>%
   mutate(across(b:d, list(zscore = ~as.numeric(scale(.)))))

Réponses

4 RonakShah Aug 18 2020 at 05:28

Vous pouvez utiliser utiliser selectpour sélectionner les colonnes qui se terminent par "zscore"et utiliser rowSums:

library(dplyr)
df1 %>%
  group_by(a) %>%
  mutate(across(b:d, list(zscore = ~as.numeric(scale(.))))) %>%
  ungroup %>%
  mutate(total = rowSums(select(., ends_with('zscore'))))

# A tibble: 30 x 8
#   a         b     c     d b_zscore c_zscore d_zscore  total
#   <chr> <dbl> <dbl> <dbl>    <dbl>    <dbl>    <dbl>  <dbl>
# 1 a      7.17 14.8   8.45    0.697   0.101    0.0179  0.816
# 2 a      7.42 19.7   3.97    0.841   1.17    -1.14    0.865
# 3 a      5.78 19.2   9.66   -0.108   1.05     0.332   1.28 
# 4 a      5.09 17.7  12.8    -0.508   0.732    1.14    1.36 
# 5 a      7.21 12.9   6.24    0.721  -0.329   -0.555  -0.163
# 6 a      2.36 13.7   2.50   -2.09   -0.146   -1.52   -3.76 
# 7 a      7.26 10.9  10.7     0.749  -0.774    0.593   0.567
# 8 a      5.45  6.18 12.8    -0.302  -1.80     1.14   -0.965
# 9 b      5.43 18.2   9.55   -0.445   1.12     1.34    2.02 
#10 b      4.16 12.1   4.11   -1.06    0.0776  -1.02   -2.01 
# … with 20 more rows
2 Wimpel Aug 18 2020 at 06:30

voici une solution data.table

Il fait essentiellement la même chose que le code de la réponse de Ronak, mais ensuite dans la data.tablesyntaxe.

explication
setDT(df1_z)est utilisée pour définir df1_zun format data.table
total := rowSums(.SD)crée une nouvelle colonne total, qui avait la valeur rowSumsdu .SD(un ensemble de colonnes sélectionnées)
.SDcols = patterns("_zscore$")définit les colonnes sélectionnées pour .SD. Ici, ce sont les colonnes dont le nom correspond au modèle regex _zscore$(ce qui signifie: se terminant par _zscore)

library( data.table )
setDT(df1_z)[, total := rowSums(.SD), .SDcols = patterns("_zscore$")]