问题描述
我想知道如何遍历这个列表而不包含两次相同的数字。
import itertools
def sum_pairs(ints,s):
indexes = []
pair = []
for numbers in itertools.combinations(ints,2):
if sum(numbers) == s:
pair.append(numbers)
for n in numbers:
indexes.append(ints.index(n))
print(pair)
print(indexes)
a = [10,5,2,3,7,5]
target = 10
输出如下:
[(5,5),(3,7)]
[1,1,4]
'pair' 正确输出 5 和 5 等于 10,但是当我使用变量 'indexes' 检查数字的来源时,我可以看到相同的 5 被使用了两次,而第二个 5 从未被考虑在内.我正在寻找的是如何修改它以在相同的索引中不添加相同的数字两次。例如。索引的输出将是 [1,4]。
非常感谢。
解决方法
改为在索引上运行组合。顺便说一句,您的索引定义不那么常见。如果您明白我的意思,请尝试使用下面的附加内容更改“扩展”
def sum_pairs(ints,s):
indexes = []
pair = []
for numbers in itertools.combinations(range(len(ints)),2):
if ints[numbers[0]]+ints[numbers[1]] == s:
indexes.extend(numbers)
pair.append((ints[numbers[0]],ints[numbers[1]]))
print(pair)
print(indexes)
,
使用 enumerate
走私每个值的索引:
import itertools
def sum_pairs(ints,s):
indexes = []
pair = []
for (ix,x),(iy,y) in itertools.combinations(enumerate(ints),2):
if x + y == s:
pair.append((x,y))
indexes += (ix,iy)
print(pair)
print(indexes)
a = [10,5,2,3,7,5]
target = 10
sum_pairs(a,target)
输出:
[(5,5),(3,7)]
[1,4]
为了简化值的使用,我将 tuple
的 tuple
解包为名称(x
和 y
是“真实”值,{{1 }} 是 ix
的索引,x
是 iy
的索引)。通过将索引附加到值,您始终可以准确地知道它的来源,而无需猜测。
您对 y
方法的使用不起作用,因为实际上,您输入中的两个 index
是无法区分的(在 CPython 上,由于 5
的小优化,它们实际上是同一个对象),而 int
只返回它找到的第一个(并且每次都必须不必要地重新扫描)。通过将索引与关联值保留在一起,您根本不必重新检查,您已经知道了。