python - 在numpy数组中查找非零之前的零数

Question

我有一个 numpy 数组 A。我想以一种有效的方式返回 A 中非零之前的零数，因为它在循环中。

如果 A = np.array([0,1,2]) 则 np.nonzero(A)[0][0] 返回 1。但是如果A = np.array([0,0,0]) 这行不通(在这种情况下我想要答案 3)。而且，如果 A 非常大并且第一个非零值接近开头，这似乎效率很低。

Answer 1

通过在数组末尾添加一个非零数字，您仍然可以使用 np.nonzero 来获得您想要的结果。

A = np.array([0,1,2])
B = np.array([0,0,0])

np.min(np.nonzero(np.hstack((A, 1))))   # --> 1
np.min(np.nonzero(np.hstack((B, 1))))   # --> 3