Список Python: если пара чисел равна 0 возвращает позиции элементов

Question

Я пытаюсь написать функции python, которые принимают в списке или массиве. если любые два числа в списке или массиве равны нулю, я хочу вернуть позиции этих двух чисел.

Вот что у меня есть. Я прохожу через список и имею вложенный цикл из того же списка, чтобы проверить, есть ли внешний итератор плюс вложенный итератор == 0: если да, то я пытаюсь вернуть позиции двух чисел, которые == 0. но это просто вместо этого дает мне фактические два числа. Странно, что результат ниже (хотя это не то, что я хочу) должен быть (3, -3). Любой вклад будет высоко оценен

def twosum(nums): 
    for x in nums: 
     for y in nums: 
      if x + y == 0: 
       return nums[x],nums[y] 
print twosum([1, 3, 5, -3]) 
output = (-3, 3) 

Answer 1

Вы должны использовать enumerate() в то время как итерацию по списку, как:

def twosum(nums): 
    for i, x in enumerate(nums): 
     for j, y in enumerate(nums): 
      if x + y == 0: 
       return i, j 

Однако эта логика может быть дополнительно оптимизирована как:

def twosum(nums): 
    for i, x in enumerate(nums): 
     if -x in nums[i:]: # check '-'ive of the number in non-iterated list 
      return x, -x 

Answer 2

Вы бы return nums.index(x), nums.index(y)