2016-10-14 4 views
0

Я не уверен, что правильная терминология для этого, но я хотел бы объединить два массива, так что полученный массив в паре каждый элемент из А с каждым элементом из B:объединить массивы numpy по элементам?

A = [1, 2, 3] 
B = [1, 2, 3] 

result = [[1,1], 
      [1,2], 
      [1,3], 
      [2,1], 
      [2,2], 
      [2,3], 
      [3,1], 
      [3,2], 
      [3,3]] 

Есть ли NumPy метод для этого или мне нужно создать цикл for и построить целый новый массив?

+0

Не в NumPy, но в пределах itertools (itertools.product). – sascha

+0

Связано: [Numpy: декартово произведение массивов x и y в единый массив двумерных точек] (http://stackoverflow.com/questions/11144513/numpy-cartesian-product-of-x-and-y-array- точки-в-одного массива из-2d-точек) –

ответ

2

Генерировать, что с помощью:

[(x,y) for x in A for y in B]