如何在NumPy中按字典序生成无重复无逆序的数组两两组合

你需要的是数组元素的无序二元无重复组合（对应数学上的组合数$C(n,2)$结果），以下是几种可直接运行的实现：

方案1：全版本兼容的高性能实现（推荐） #

通过上三角索引直接生成有效配对，无冗余计算，大数组场景下性能最优：

import numpy as np
array = np.array([0,1,2,3])

# 取偏移1位的上三角索引，天然满足i<j，自动排除同元素、逆序无效配对
i, j = np.triu_indices(len(array), k=1)
combinations = np.stack([array[i], array[j]], axis=1)
print(combinations)

输出结果：

[[0 1]
 [0 2]
 [0 3]
 [1 2]
 [1 3]
 [2 3]]

np.triu_indices(n, k=1)会返回n阶方阵中主对角线以上、偏移1位区域的行列索引对，所有索引对均满足行号小于列号，从根源上避免了两类无效结果：

• 行号等于列号时的同元素配对（如[0,0]、[1,1]）
• 行号大于列号时的逆序配对（如[1,0]、[3,2]）

方案2：基于原有meshgrid逻辑改造 #

如果要保留你原本的meshgrid实现思路，只需要在生成全量配对后增加一行过滤逻辑即可：

import numpy as np
array = np.array([0,1,2,3])

combinations = np.array(np.meshgrid(array, array)).T.reshape(-1,2)
# 仅保留第一个元素小于第二个元素的配对
combinations = combinations[combinations[:,0] < combinations[:,1]]

该方案写法直观，但会先生成$n^2$个全量配对再过滤掉近一半的无效数据，数组长度较大时内存和计算开销更高，仅适合小数据量场景。

方案3：高版本NumPy原生API #

NumPy 1.25.0及以上版本内置了组合生成函数，代码最简洁：

import numpy as np
array = np.array([0,1,2,3])

combinations = np.lib.arraysetops.combinations(array, 2)

输出和上述方案完全一致。

内容的提问来源于stack exchange，提问作者Armando Bridena