生成從2個numpy的矢量

Question

數對陣列是否有numpy的一個簡單的方法而無需循環，以產生數對從2門維numpy的陣列（載體）的陣列？

輸入：

a = [1, 2, 3] 
b = [4, 5, 6] 

輸出：

c = [(1,4), (1,5), (1,6), (2,4), (3,5), (2,6), (3,4), (3,5), (3,6)] 

我想知道如果有，做類似的東西，以這樣的功能：

c = [] 
for i in range(len(a)): 
    for j in range(len(b)): 
     c.append((a[i], b[j])) 

Answer 1

您可以使用itertools.product此：

from itertools import product 

c = list(product(a, b)) 

這給：在`C = [（I，J），其中i：

c == [(1, 4), (1, 5), (1, 6), (2, 4), (2, 5), (2, 6), (3, 4), (3, 5), (3, 6)]