如何修复代码中的输出问题？

Question

So I wrote this code to print out true or false depending on whether the value goes below zero or not. Here is my code

    L = [10,10,10]
init_hp = 0
def is_dead(L: list[int], init_hp:int):
    if init_hp == 0:
        return True
    elif sum(L) + init_hp > 0:
        return False
    else:
        return True
print(is_dead(L,init_hp))

However, for the L: list I want to make sure that the output prints out True if the value has gone below zero already. For example, if L: list = [2,-3,5] , I want to make sure that the output prints out True , because 2+(-3), the first two variables, already return a negative number,"-1", EVEN THOUGH THE END RESULT IS POSITIVE... however using my code, that doesn't work, what code should I use to make sure that it will print true if a scenario like this happens

Answer 1

You can use loop and iterate all numbers in the list.

L = [10, 10, 10]
init_hp = 0
def is_dead(L: list[int], init_hp:int):
    if init_hp == 0:
        return True
    v = 0
    for num in L:
        if v+num < 0:
            return True
        v += num
    
    if init_hp+v < 0:
         return True
    return False

print(is_dead(L,init_hp))

Answer 2

The shortest way I could think of was to use itertools.accumulate :

from typing import List
from itertools import accumulate


def is_dead(lst: List[int], i_hp: int):
    if i_hp == 0:
        return True
    for total in accumulate(lst):
        if total < 0:
            return True
    return False


print(is_dead([2, -3, 5], 1))
# True
print(is_dead([10, 10, 10], 1))
# False

Or slightly shorter (but also may be slightly less memory efficient):

def is_dead(lst: List[int], i_hp: int):
    if i_hp == 0:
        return True
    if [True for total in accumulate(lst) if total < 0]:
        return True
    return False

Sources:

• itertools.accumulate docs

Answer 3

You can use a for loop to iterate through the numbers and check whether the sum is less than zero or not.

def is_dead(L: list[int], init_hp: int):
    if init_hp <= 0:
        return True

    s = 0
    for num in L:
        s += num
        if s < 0:
            return True

    return False  # sum > 0


# Test cases:
print(is_dead([2, -3, 5], 1))  # True
print(is_dead([10, 10, 10], -1))  # True
print(is_dead([10, 10, 10], 1))  # False