python dataframeで1行ずつの処理をしたい

Question

というdataframeのvolを、taker_sideのbuy、sellによって新しく振り分けたいと思っています。欲しい完成図は下記のようなものです。

やったこと

df = dataframe
for i in df["taker_side"]:
    if i == "buy":
        df["vol_buy"] = df["vol"]
    if i == "sell":
        df["vol_sell"] = df["vol"]

結果
![]

vol_buyとvol_sellの両方にvolの値がそのまま入ってしまいます。
taker_sideのbuy、sellでvolの値をそれぞれvol_buyとvol_sellに振り分けたいのですが、どうすればいいでしょうか。

Answer 1

こういう感じです。

>>> print(df)
     vol taker_side
0  0.030          b
1  0.010          b
2  0.040          s
3  0.002          s
4  0.002          b
>>>
>>> df['vol_buy'] = df['vol'].where(df['taker_side']=='b', 0)
>>> df['vol_sell'] = df['vol'].where(df['taker_side']=='s', 0)
>>>
>>> print(df)
     vol taker_side  vol_buy  vol_sell
0  0.030          b    0.030     0.000
1  0.010          b    0.010     0.000
2  0.040          s    0.000     0.040
3  0.002          s    0.000     0.002
4  0.002          b    0.002     0.000

Comments

[K.Andou]

まさに求めていたコードです！ありがとうございます1