Python从列表中找到两个数字的总和

时间:2016-03-02 16:52:52

标签: python if-statement for-loop sum

我正在尝试编写一些代码来检查数字是否是列表/字典中任意两个数字的总和,如果找到一个数字,它就会停止运行。但是,我遇到了一些错误。也许我的逻辑错了,但这是我的代码:

a = [1,2,3,4,5,6,7,8,9]

randomNumber = 8

print len(a)
length_of_a = len(a)
for first in range(0,length_of_a -1):
    aa = a
    bb = a
    del bb[first]
    length_of_b = len(bb)
    print bb, length_of_b
    for second in range(0, length_of_b-1):
        print aa[first], bb[second]
        x = aa[first] + bb[second]
        print x
        if x == randomNumber:
            print "Sum Found!"
            break
        else:
            print "No Sum"

所以我的错误:

  • aa [first]不是从1开始
  • 当第二个数组的大小始终为8时,我的第二个数组会变小。因此,给我错误“IndexError:list index out of range”
  • 找到一笔款项时不会停止

任何帮助都会很棒

1 个答案:

答案 0 :(得分:3)

有一种非常简单的方法可以做你想要的。它是一行,因此您可以使用lambda函数定义它:

is_sum = lambda seq, x: any(x == y + z for yi, y in enumerate(seq) for zi, z in enumerate(seq) if zi != yi)

使用:

>>> is_sum([1, 2, 3, 4], 5)
True
>>> is_sum([1, 2, 3, 4], 6)
True
>>> is_sum([1, 2, 3, 4], 12)
False