python - 如何执行计算比较每个元素与 numpy/scipy 中的每个其他元素

Question

我有一个 2D 数组，我需要对其进行一些分析，其中包括对每对可能的元素进行计算，然后将它们相加。

问题是我需要避免为每对计算两次 - 如果我正在查看元素 A 和 B，我需要对 A*B 进行计算并避免对 B*A 重复计算。

目前我正在这样做：

comb=[]
amo=len(inds)
for m in range(0,amo):
    for n in range(m+1,amo):
        comb.append([inds[m],inds[n]])

哪里inds是一维数组。

Answer 1

看起来您正在寻找itertools.combinations。

from itertools import combinations
comb = list(combinations(inds, 2))