python - 返回列表中最后一个非零元素的索引

Question

我得到以下格式的数据:

[-2, -2, 0, 0, 0, 0, 0]
[-2, 20, -1, 0, 3, 0, 0]

每一行都是不同的输入。列表可能超过 7 个元素。我需要返回最后一个非零元素的索引位置，所以:

[-2, -2, 0, 0, 0, 0, 0]
>>> 1
[-2, 20, -1, 0, 3, 0, 0]
>>> 4

大多数情况下，以下代码会执行此操作:

def getIndex(list):
    for element in reversed(list):
        if element != 0:
            return list.index(element)

但是，当有两个相同的数字时，它不起作用，如上面的第一个示例，它返回 0，因为 -2 在两个列表的第 0 位和第 1 位。

那么如何获取列表中最后一个非零元素的索引，即使存在具有相同值的元素？

Answer 1

这个问题被标记为 numpy 但我没有看到明显的解决方案:

numpy.max(numpy.nonzero(list))

示例提供的列表:

>>> import numpy as np
>>> np.max(np.nonzero([-2, -2, 0, 0, 0, 0, 0]))
1
>>> np.max(np.nonzero([-2, 20, -1, 0, 3, 0, 0]))
4