Уникальное сочетание всех элементов из двух (или более) векторов

97

Я пытаюсь создать уникальную комбинацию всех элементов из двух векторов разного размера в R.

Например, первый вектор

a <- c("ABC", "DEF", "GHI")

а второй - даты, хранящиеся в настоящее время в виде строк

b <- c("2012-05-01", "2012-05-02", "2012-05-03", "2012-05-04", "2012-05-05")

Мне нужно создать фрейм данных с двумя столбцами, подобными этому

> data
    a          b
1  ABC 2012-05-01
2  ABC 2012-05-02
3  ABC 2012-05-03
4  ABC 2012-05-04
5  ABC 2012-05-05
6  DEF 2012-05-01
7  DEF 2012-05-02
8  DEF 2012-05-03
9  DEF 2012-05-04
10 DEF 2012-05-05
11 GHI 2012-05-01
12 GHI 2012-05-02
13 GHI 2012-05-03
14 GHI 2012-05-04
15 GHI 2012-05-05

По сути, я ищу уникальную комбинацию, рассматривая все элементы одного вектора (a), сопоставленные со всеми элементами второго вектора (b).

Идеальным решением было бы обобщение на большее количество входных векторов.


См. Также:
Как создать матрицу комбинаций

Годель
источник

Ответы:

141

это может быть то, что вам нужно

> expand.grid(a,b)
   Var1       Var2
1   ABC 2012-05-01
2   DEF 2012-05-01
3   GHI 2012-05-01
4   ABC 2012-05-02
5   DEF 2012-05-02
6   GHI 2012-05-02
7   ABC 2012-05-03
8   DEF 2012-05-03
9   GHI 2012-05-03
10  ABC 2012-05-04
11  DEF 2012-05-04
12  GHI 2012-05-04
13  ABC 2012-05-05
14  DEF 2012-05-05
15  GHI 2012-05-05

Если полученный порядок не соответствует вашему желанию, вы можете отсортировать его позже. Если вы назовете аргументы в expand.grid, они станут именами столбцов:

df = expand.grid(a = a, b = b)
df[order(df$a), ]

И expand.gridобобщается на любое количество входных столбцов.

охота на кроликов
источник
4
И без необходимости plyrпросто делать сортировку:result <- expand.grid(a=a,b=b); result <- result[order(result$a,result$b),];
thelatemail
Может ли кто-нибудь с большим количеством представителей, чем я, принять этот ответ?
Джош
Если порядок и имена должны быть такими, как в вопросе:expand.grid(b=b,a=a)[2:1]
GKi
Обратите внимание, что заголовок «Уникальные комбинации» - этот ответ решает проблему OP, но если 2 столбца имеют один и тот же тип данных и вы применяете expand.grid, у вас будут уникальные перестановки, а не уникальные комбинации
Брент
29

tidyrПакет обеспечивает хорошую альтернативу crossing, которая работает лучше , чем классические expand.gridфункции , потому что (1) строки не превращаются в факторы , и (2) сортировка более интуитивная:

library(tidyr)

a <- c("ABC", "DEF", "GHI")
b <- c("2012-05-01", "2012-05-02", "2012-05-03", "2012-05-04", "2012-05-05")

crossing(a, b)

# A tibble: 15 x 2
       a          b
   <chr>      <chr>
 1   ABC 2012-05-01
 2   ABC 2012-05-02
 3   ABC 2012-05-03
 4   ABC 2012-05-04
 5   ABC 2012-05-05
 6   DEF 2012-05-01
 7   DEF 2012-05-02
 8   DEF 2012-05-03
 9   DEF 2012-05-04
10   DEF 2012-05-05
11   GHI 2012-05-01
12   GHI 2012-05-02
13   GHI 2012-05-03
14   GHI 2012-05-04
15   GHI 2012-05-05
гипотеза
источник
13

Отсутствует в этом Обзор - это CJфункция из-пакет. С помощью:

library(data.table)
CJ(a, b, unique = TRUE)

дает:

      a          b
 1: ABC 2012-05-01
 2: ABC 2012-05-02
 3: ABC 2012-05-03
 4: ABC 2012-05-04
 5: ABC 2012-05-05
 6: DEF 2012-05-01
 7: DEF 2012-05-02
 8: DEF 2012-05-03
 9: DEF 2012-05-04
10: DEF 2012-05-05
11: GHI 2012-05-01
12: GHI 2012-05-02
13: GHI 2012-05-03
14: GHI 2012-05-04
15: GHI 2012-05-05

ПРИМЕЧАНИЕ: начиная с версии 1.12.2 автоматически CJназываются результирующие столбцы (см. Также здесь и здесь ).

Яап
источник
4

Начиная с версии 1.0.0, tidyrпредлагает собственную версию expand.grid(). Он завершает существующую семью expand(), nesting()и crossing()с функцией низкого уровня , который работает с векторами .

По сравнению с base::expand.grid():

Изменяет первый элемент быстрее всего. Никогда не преобразует строки в множители. Не добавляет никаких дополнительных атрибутов. Возвращает тиббл, а не фрейм данных. Может расширять любой обобщенный вектор, включая фреймы данных.

a <- c("ABC", "DEF", "GHI")
b <- c("2012-05-01", "2012-05-02", "2012-05-03", "2012-05-04", "2012-05-05")

tidyr::expand_grid(a, b)

   a     b         
   <chr> <chr>     
 1 ABC   2012-05-01
 2 ABC   2012-05-02
 3 ABC   2012-05-03
 4 ABC   2012-05-04
 5 ABC   2012-05-05
 6 DEF   2012-05-01
 7 DEF   2012-05-02
 8 DEF   2012-05-03
 9 DEF   2012-05-04
10 DEF   2012-05-05
11 GHI   2012-05-01
12 GHI   2012-05-02
13 GHI   2012-05-03
14 GHI   2012-05-04
15 GHI   2012-05-05
tmfmnk
источник
3

вы можете использовать функцию заказа для сортировки любого количества столбцов. для вашего примера

df <- expand.grid(a,b)
> df
   Var1       Var2
1   ABC 2012-05-01
2   DEF 2012-05-01
3   GHI 2012-05-01
4   ABC 2012-05-02
5   DEF 2012-05-02
6   GHI 2012-05-02
7   ABC 2012-05-03
8   DEF 2012-05-03
9   GHI 2012-05-03
10  ABC 2012-05-04
11  DEF 2012-05-04
12  GHI 2012-05-04
13  ABC 2012-05-05
14  DEF 2012-05-05
15  GHI 2012-05-05

> df[order( df[,1], df[,2] ),] 
   Var1       Var2
1   ABC 2012-05-01
4   ABC 2012-05-02
7   ABC 2012-05-03
10  ABC 2012-05-04
13  ABC 2012-05-05
2   DEF 2012-05-01
5   DEF 2012-05-02
8   DEF 2012-05-03
11  DEF 2012-05-04
14  DEF 2012-05-05
3   GHI 2012-05-01
6   GHI 2012-05-02
9   GHI 2012-05-03
12  GHI 2012-05-04
15  GHI 2012-05-05`
Изан
источник