python - 压缩不均匀的 numpy 数组

Question

考虑以下 numpy.arrays:

a = np.array([1., 2., 3.])
b = np.array([4., 5.])
c = np.array([6., 7.])

我需要结合这些，所以我最终得到以下内容:

[(1., 4., 6.), (1., 5., 7.), (2., 4., 6.), (2., 5., 7.), (3., 4., 6.), (3., 5., 7.)]

请注意，在这种情况下，数组 a 恰好是最大的数组。然而，这并不能保证。也不能保证长度。换句话说，任何数组都可以是最长的，并且每个数组的长度都是任意的。

我尝试使用 itertools.izip_longest 但我只能将 fillvalue 用于带有 3. 的元组，这是行不通的。我也尝试了 itertools.product，但我的结果不是真正的笛卡尔积。

Answer 1

您可以转置 b 和 c，然后使用 itertools.product< 创建 a 与转置数组的乘积:

>>> from itertools import product
>>> [np.insert(j,0,i) for i,j in product(a,np.array((b,c)).T)]
[array([ 1.,  4.,  6.]), array([ 1.,  5.,  7.]), array([ 2.,  4.,  6.]), array([ 2.,  5.,  7.]), array([ 3.,  4.,  6.]), array([ 3.,  5.,  7.])]
>>>