我有一個列表[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]但似乎效率很低。有沒有更有效的方法來做同樣的事情？
1 回答
莫回無
TA貢獻1865條經驗 獲得超7個贊
既然你說順序并不重要，那么你可以通過先對數字進行排序來大大簡化問題。
>>> 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)]
添加回答
舉報
0/150
提交
取消