Как перебирать имена столбцов в r (с примерами)

Часто вам может потребоваться перебрать имена столбцов фрейма данных в R и выполнить определенные операции над каждым столбцом. Есть два распространенных способа сделать это:

Метод 1: используйте цикл For

 for (i in colnames(df)){
   some operation
}

Способ 2: используйте sapply()

 sapply(df, some operation )

В этом руководстве показан пример использования каждого из этих методов на практике.

Метод 1: используйте цикл For

Следующий код показывает, как перебирать имена столбцов фрейма данных с помощью цикла for и отображать среднее значение каждого столбца:

 #create data frame
df <- data.frame(var1=c(1, 3, 3, 4, 5),
                 var2=c(7, 7, 8, 3, 2),
                 var3=c(3, 3, 6, 6, 8),
                 var4=c(1, 1, 2, 8, 9))

#view data frame
df

  var1 var2 var3 var4
1 1 7 3 1
2 3 7 3 1
3 3 8 6 2
4 4 3 6 8
5 5 2 8 9

#loop through each column and print mean of column
for (i in colnames(df)){
    print( mean (df[[i]]))
}

[1] 3.2
[1] 5.4
[1] 5.2
[1] 4.2

Способ 2: используйте sapply()

Следующий код показывает, как перебирать имена столбцов фрейма данных с помощью sapply() и отображать среднее значение каждого столбца:

 #create data frame
df <- data.frame(var1=c(1, 3, 3, 4, 5),
                 var2=c(7, 7, 8, 3, 2),
                 var3=c(3, 3, 6, 6, 8),
                 var4=c(1, 1, 2, 8, 9))

#view data frame
df

  var1 var2 var3 var4
1 1 7 3 1
2 3 7 3 1
3 3 8 6 2
4 4 3 6 8
5 5 2 8 9

#loop through each column and print mean of column
sapply(df, mean )

var1 var2 var3 var4 
 3.2 5.4 5.2 4.2

Обратите внимание, что оба метода возвращают одинаковые результаты.

Связанный: Руководство по применению(), lapply(), sapply() и Tapply() в R.