Блог

Я пытаюсь использовать R для создания функции, которая принимает два списка в качестве входных данных и возвращает общие элементы, но без использования функции intersect. Я все еще борюсь с некоторыми вещами, когда изучаю R:

• списки против векторов против фреймов данных
• как сохранить / сохранить результаты из циклов и функций for
• как использовать sapply / mapply / etc.

Вот что у меня есть на данный момент. Сначала я думал, что это сработало, когда единственным общим элементом было ‘5’, но когда я добавил более общие числа, он по-прежнему возвращает только ‘5’.

** Примечание — после написания этого я понял, что необходимо запускать это только в одном направлении (есть ли что-нибудь в list_a также в list_b), но теперь мне любопытно, как запустить эту версию с точки зрения программирования.

 a <- list(1,2,3,4,5)
b <- list(3,4,5,6,7,8,9,10,11)

a_match <- numeric()
b_match <- numeric()

ab <- function(list_a, list_b) {
  for (i in list_a) {
    results <- if(i %in% list_b) {i}
    a_in_b <- c(a_match, results)
  }
  for (i in list_b) {
    results <- if(i %in% list_a) {i}
    b_in_a <- c(b_match, results)
  }
  combined <- c(a_in_b, b_in_a)
  unique(combined)
}


ab(a, b)
  

Есть ли способ использовать ‘apply’ здесь? и:

1. возвращает вектор, длина которого отличается от длины входных данных (ограничение sapply)
2. может принимать несколько входных данных разной длины (ограничение отображения)

Ваши a_match и b_match находятся вне объявленных вне функции. Я исправил ваш пример проверки ниже:

 a <- list(1,2,3,4,5)
b <- list(3,4,5,6,7,8,9,10,11)

ab <- function(list_a, list_b) {
  
  a_match <- numeric()
  b_match <- numeric()

  
  for (i in list_a) {
    results <- if(i %in% list_b) {i}
    a_match <- c(a_match, results)
  }
  
  for (i in list_b) {
    results <- if(i %in% list_a) {i}

    b_match <- c(b_match, results)
  }
  combined <- c(a_match,b_match)
  
  unique(combined)
}


ab(a, b)
  

мы могли бы просто использовать %in% функцию для получения элементов, которые находятся в a и b => a %in% b является логическим вектором, который мы используем для подмножества исходного списка.

 a %in% b
#> [1] FALSE FALSE  TRUE  TRUE  TRUE
unique(a[a %in% b])
  

 [[1]]
[1] 3

[[2]]
[1] 4

[[3]]
[1] 5
  

Имейте в виду, что большинство операторов (функций) в базовом векторизованы, поэтому нет необходимости использовать семейство apply в этом.

 unlist(unique(a[a %in% b]))
#> [1] 3 4 5