Cumsum-Reset und NaN ⇐ Python

[Guest]

Wenn ich eine pandas.core.series.Series namens ts mit entweder Einsen oder NaNs wie folgt habe:

Code: Select all

3382   NaN
3381   NaN
...
3369   NaN
3368   NaN
...
15     1
10   NaN
11     1
12     1
13     1
9    NaN
8    NaN
7    NaN
6    NaN
3    NaN
4      1
5      1
2    NaN
1    NaN
0    NaN

Ich würde gerne die Summe dieser Serie berechnen, aber sie sollte an der Position der NaNs wie folgt zurückgesetzt (auf Null gesetzt) ​​werden:

Code: Select all

3382   0
3381   0
...
3369   0
3368   0
...
15     1
10     0
11     1
12     2
13     3
9      0
8      0
7      0
6      0
3      0
4      1
5      2
2      0
1      0
0      0

Idealerweise hätte ich gerne eine vektorisierte Lösung!

Ich habe jemals eine ähnliche Frage mit Matlab gesehen: Matlab Cumsum bei NaN zurückgesetzt?

aber ich weiß nicht, wie ich diese Zeile übersetzen soll d = diff([0 c(n)]);