获取列表中所有可能相差 1 的数字对

Question

我有一个列表[0, 3, 2, 1]，我想找到差为 1 的 2 个数字的所有可能组合，即|a - b| = 1。所以期望的结果是[(0, 1), (3, 2), (2, 1)]

我知道itertools.combinations并且我知道我可以这样做：

nums = [0, 3, 2, 1]

combs = list(combinations(nums, 2))

adj_combs = [comb for comb in combs if abs(comb[0] - comb[1]) == 1]

但似乎效率很低。有没有更有效的方法来做同样的事情？

Answer 1

既然你说顺序并不重要，那么你可以通过先对数字进行排序来大大简化问题。>>> nums = [0, 3, 2, 1]>>> sorted_nums = sorted(nums)>>> adj_combs = [(a,b) for a,b in zip(sorted_nums,sorted_nums[1:]) if abs(a-b) == 1]>>> adj_combs[(0, 1), (1, 2), (2, 3)]