<分区>
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]))
最佳答案
您可以使用 itertools.product为此:
from itertools import product
c = list(product(a, b))
这给出:
c == [(1, 4), (1, 5), (1, 6), (2, 4), (2, 5), (2, 6), (3, 4), (3, 5), (3, 6)]
关于python - 从 2 个 numpy 向量生成数字对数组，我们在Stack Overflow上找到一个类似的问题： https://stackoverflow.com/questions/21651980/