Pythonn维数组组合

Question

假设任意数量的任意长度的数组。我想从数组中的值构造所有组合的 n 维数组。或者更好的是，所有组合的列表。

但是，我也想要每个组合中的前一个 "diagonal" 元素，除非这样的元素不存在，在这种情况下，不存在的值设置为 -inf.

以前者为例。以下简单的二维案例：

v1=[-2,2]
v2=[-3,3]

我将从中得到所有的组合

[[-2,-3],  
 [-2,3],  
 [2,-3],  
 [2,3]]

或者二维数组/矩阵形式

      -3     3
-2  -2,-3  -2,3
 2   2,-3   2,3

现在我还想要一个新列，每个元素都包含以前的 "diagonal" 元素（在本例中只有 1 个真实的这种情况）。前面的 "diagonal" 元素是指索引 i-1、j-1、k-1、...、n-1 处的元素。在空白处，我们采用所有可能的先前值。

  1      2
-2,-3  -inf,-inf
-2, 3  -inf,-3
 2,-3  -2,-inf
 2, 3  -2,-3

编辑：这里是2D情况的代码，对于一般的n-case用处不大

import math

v1=[-3,-1,2,4]
v2=[-2,0,2]

tmp=[]
tmp2=[]
for i in range(0,len(v1)):
    for j in range(0,len(v2)):
        tmp.append([v1[i],v2[j]])
        if i==0 and j==0:
            tmp2.append([-math.inf,-math.inf])
        elif i==0:
            tmp2.append([-math.inf,v2[j-1]])
        elif j==0:
            tmp2.append([v1[i-1],-math.inf])
        else:
            tmp2.append([v1[i-1],v2[j-1]])

等等

tmp

[[-3, -2],
 [-3, 0],
 [-3, 2],
 [-1, -2],
 [-1, 0],
 [-1, 2],
 [2, -2],
 [2, 0],
 [2, 2],
 [4, -2],
 [4, 0],
 [4, 2]]

和

tmp2

[[-inf, -inf],
 [-inf, -2],
 [-inf, 0],
 [-3, -inf],
 [-3, -2],
 [-3, 0],
 [-1, -inf],
 [-1, -2],
 [-1, 0],
 [2, -inf],
 [2, -2],
 [2, 0]]

Answer 1

看看itertools.product()。

要获得 "diagonals"，您可以取向量索引的乘积而不是向量本身。这样您就可以访问每个组合的值以及组合的先前值。

示例：

import itertools

v1=[-2,2]
v2=[-3,3]
vectors = [v1, v2]

combs = list(itertools.product(*[range(len(v)) for v in vectors]))

print(combs)

[(0, 0), (0, 1), (1, 0), (1, 1)]

print([[vectors[vi][ci] for vi, ci in enumerate(comb)] for comb in combs])

[[-2, -3], [-2, 3], [2, -3], [2, 3]]

print([[(vectors[vi][ci-1] if ci > 0 else np.inf) for vi, ci in enumerate(comb)] for comb in combs])

[[inf, inf], [inf, -3], [-2, inf], [-2, -3]]

Pythonn维数组组合

Python n-dimensional array combinations

python

numpy

itertools

n-dimensional

pandas