python pandas 累計和(符号違いで0にリセット)

Question

###前提・実現したいこと
pythonでpandasを使用しています。
-1, 0, 1 しかないnumというcolumnがあり、
numの符号が一致している限り、累積和を計算し、
符号が異なる場合、0にリセットして、
新しい列resultに格納したいと考えています。

###発生している問題・エラーメッセージ
エラーは出ませんが、期待する出力結果になりません。

エラーメッセージ

###該当のソースコード

df = pd.DataFrame({"num": [0, 1, 1, -1, 1, -1, -1, -1, 1]})
df["result"] = 0
df.ix[(df["num"] > 0) & (df["result"].shift(1) >= 0), "result"] = df["result"].shift(1) + 1
df.ix[(df["num"] < 0) & (df["result"].shift(1) <= 0), "result"] = df["result"].shift(1) - 1
print(df)

###試したこと
上記のソースコードです。cumsum()を上手く使えばできるのでしょうか？
期待する出力結果：
{result: [0, 1, 2, 0, 1, 0, -1, -2, 0]}

###補足情報(言語/FW/ツール等のバージョンなど)
Win10 x64
Python3.6 x64

Answer 1

groupby()とcumsum()を使ってスマートに記述する方法がある気がするのですが・・残念ながら考え付きませんでした。スミマセン

で、

ここでは、シンプルにiterrows()で行毎にループ処理するのでは駄目でしょうか。
すこしやっつけですが、以下のように記述できると思います。

import pandas as pd

df = pd.DataFrame({"num": [0, 1, 1, -1, 1, -1, -1, -1, 1]})

current = 1
total = 0
for idx,row in df.iterrows():
    if -current == row['num']:
        current = row['num']
        df.ix[idx,'result'] = total = 0
    else:
        df.ix[idx,'result'] = total = row['num'] + total

Comments

[draftnyan]

magichanさん、ご回答ありがとうございます。
パフォーマンスが要求されそうな処理だったので、forで回すことは最初から除外していました...
問題がなければ、そのまま採用させていただきます。
ありがとうございました。