2016-10-14 164 views
0

我不知道正確的術語是什麼,但我想兩個數組,每個項目的B相結合使得產生的陣列已經配對從A每個項目:結合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數組點到2D單點陣列中的笛卡爾乘積](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]