Как использовать функцию which в r (с примерами)

Функция Which() в R возвращает положение элементов логического вектора, имеющих значение TRUE .

В этом руководстве представлено несколько примеров практического использования этой функции.

Пример 1. Поиск элементов в векторе

Следующий код показывает, как найти положение всех элементов вектора, равного 5:

 #create data
data <- c(1, 2, 2, 3, 4, 4, 4, 5, 5, 12)

#find the position of all elements equal to 5
which (data == 5)

[1] 8 9

Мы видим, что элементы в позициях 8 и 9 в векторе равны значению 5.

Мы также можем найти положение всех элементов вектора, которые не равны 5:

 #find the position of all elements not equal to 5
which (data != 5)

[1] 1 2 3 4 5 6 7 10

Мы также можем узнать, какие элементы находятся между двумя значениями или вне двух значений:

 #find the position of all elements with values between 2 and 4
which (data >= 2 & data <= 4)

[1] 2 3 4 5 6 7

#find the position of all elements with values outside of 2 and 4
which (data < 2 | data > 4)

[1] 1 8 9 10

Пример 2. Подсчет вхождений в векторе

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

 #createdata
data <- c(1, 2, 2, 3, 4, 4, 4, 5, 5, 12)

#find number of elements greater than 4
length ( which (data > 4))

[1] 3

Мы видим, что в этом векторе есть 3 элемента со значениями больше 4.

Пример 3. Поиск строк во фрейме данных

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

 #create data frame
df <- data. frame (x = c(1, 2, 2, 3, 4, 5),
                 y = c(7, 7, 8, 9, 9, 9),
                 z = c('A', 'B', 'C', 'D', 'E', 'F'))

#view data frame
df

  X Y Z
1 1 7 A
2 2 7 B
3 2 8 C
4 3 9 D
5 4 9 E
6 5 9 F

#return row that contains the max value in column x
df[ which . max (df$x), ]

  X Y Z
6 5 9 F

#return row that contains the min value in column x
df[ which . min (df$x), ]

  X Y Z
1 1 7 A

Пример 4: подмножество по строкам во фрейме данных

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

 #create data frame
df <- data. frame (x = c(1, 2, 2, 3, 4, 5),
                 y = c(7, 7, 8, 9, 9, 9),
                 z = c('A', 'B', 'C', 'D', 'E', 'F'))

#view data frame
df

  X Y Z
1 1 7 A
2 2 7 B
3 2 8 C
4 3 9 D
5 4 9 E
6 5 9 F

#return subset of data frame where values in column y are greater than 8
df[ which (df$y > 8), ]

  X Y Z
4 3 9 D
5 4 9 E
6 5 9 F

Дополнительные руководства по R можно найти на этой странице .