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 nonzerosz = a!=0### find switching pointsz[1:] ^= z[:-1]### split at switching points and discard zerosnp.split(a, *np.where(z))[1::2]# [array([1, 2, 3]), array([4, 5])]