Как создать вложенный цикл for в r (включая примеры)

Вложенный цикл for позволяет перебирать элементы в нескольких векторах (или в нескольких измерениях матрицы) и выполнять определенные операции.

Базовая структура цикла for в R:

 for (i in 1:4) {
  print (i)
}

[1] 1
[1] 2
[1] 3
[1] 4

Базовая структура вложенного цикла for такова :

 for (i in 1:4) {
  for (j in 1:2) {
    print (i*j)
  }
}

[1] 1
[1] 2
[1] 2
[1] 4
[1] 3
[1] 6
[1] 4
[1] 8

В этом руководстве показаны некоторые примеры создания вложенных циклов for в R.

Пример 1. Вложенный цикл For в R

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

 #create matrix
empty_mat <- matrix(nrow= 4 , ncol= 4 )

#view empty matrix
empty_mat
     [,1] [,2] [,3] [,4]
[1,] NA NA NA NA
[2,] NA NA NA NA
[3,] NA NA NA NA
[4,] NA NA NA NA

#use nested for loop to fill in values of matrix
for (i in 1:4) {
  for (j in 1:4) {
    empty_mat[i, j] = (i*j)
  }
}

#view matrix
empty_mat

     [,1] [,2] [,3] [,4]
[1,] 1 2 3 4
[2,] 2 4 6 8
[3,] 3 6 9 12
[4,] 4 8 12 16

Пример 2. Вложенный цикл For в R

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

 #create empty data frame
df <- data.frame(var1=c(1, 7, 4),
                 var2=c(9, 13, 15))

#view empty data frame
df

  var1 var2
1 1 9
2 7 13
3 4 15

#use nested for loop to square each value in the data frame
for (i in 1:nrow(df)) {
  for (j in 1:ncol(df)) {
    df[i, j] = df[i, j]^2
  }
}

#view new data frame
df

  var1 var2
1 1 81
2 49 169
3 16 225

Примечание о цикле

В общем, вложенные циклы for хорошо работают с небольшими наборами данных или матрицами, но они, как правило, работают довольно медленно с большими данными.

Для больших данных семейство функций приложения обычно работает намного быстрее, а пакет data.table имеет множество встроенных функций, которые эффективно работают с большими наборами данных.

Дополнительные ресурсы

Как перебирать имена столбцов в R
Как добавить строки во фрейм данных в R