python - 获取python列表中下一个非零值的索引

Question

假设您有一个列表[1,2,3,4,5,0,0,0,7,8, ...]。给定当前索引，获取下一个非零索引的最佳方法是什么？例如，如果索引为2，则下一个非零为3；如果是 4，则为 8。

Answer 1

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

def nextZeroIndex(array, start):
    nextItems = array[start+1:]
    for i, x in enumerate(nextItems):
        if x != 0:
            return 1 + start + i

print(nextZeroIndex(a, 2)) # 3
print(nextZeroIndex(a, 5)) # 8
print(nextZeroIndex(a, 4)) # 8