Python numpy运行重复的trues

Question

我想确定真值的运行总和，但是当元素为false时重置为零。

例如，输入：

np.array([0,0,1,1,1,1,0,0,0,1,0,1,0,0,1,1])

会导致：

np.array([0,0,1,2,3,4,0,0,0,1,0,1,0,0,1,2])

我更喜欢纯粹的numpy解决方案