我有一個(gè)列表[0, 3, 2, 1]，我想找到差為 1 的 2 個(gè)數(shù)字的所有可能組合，即|a - b| = 1。所以期望的結(jié)果是[(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 回答
莫回?zé)o
TA貢獻(xiàn)1865條經(jīng)驗(yàn) 獲得超7個(gè)贊
既然你說順序并不重要，那么你可以通過先對(duì)數(shù)字進(jìn)行排序來大大簡化問題。
>>> 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)]
添加回答
舉報(bào)
0/150
提交
取消