您可以使用numpy.cumsum()和numpy.argmax()。首先，计算元素的累积和。然后返回elements < 0的True/False，并返回element < 0与argmax()的第一个为True的索引。

>>> (a.cumsum() < 0).argmax()

检查两个代码：

import numpy as np

a = np.random.rand(1000)
a -= .55
a[0] = 1

def check_1(a):
    b = 0
    for i in range(len(a)):
        b += a[i]
        if b < 0:
            print(i)
            break
            
def check_2(a):
    return (a.cumsum() < 0).argmax()

输出：（生成基本随机输入）

>>> check_1(a)
6

>>> check_2(a)
6