numpy 数组中的矢量化非零组

Question

假设您有 1d numpy 数组：[0,0,0,0,0,1,2,3,0,0,0,0,4,5,0,0,0]您将如何在不使用 for 循环的情况下创建以下组？[1,2,3], [4,5]

Answer 1

这是使用的一种方法np.split：

a

# array([0, 0, 0, 0, 0, 1, 2, 3, 0, 0, 0, 0, 4, 5, 0, 0, 0])

### find nonzeros

z = a!=0

### find switching points

z[1:] ^= z[:-1]

### split at switching points and discard zeros

np.split(a, *np.where(z))[1::2]

# [array([1, 2, 3]), array([4, 5])]