fork download
  1. from itertools import accumulate
  2.  
  3. f = lambda numbers: next((i for i, (n, s) in enumerate(zip(numbers, accumulate(numbers))) if 2 * n < s), -1)
  4.  
  5. print(f([1, 2, 3, 4]))
  6. print(f([1, 2, 3]))
  7. print(f([1, 2]))
Success #stdin #stdout 0.02s 28376KB
stdin
Standard input is empty
stdout
3
-1
-1