Pandasのデータフレームで特定の条件を満たす箇所だけを書き換えたい

Question

【環境】Windows10 64bit, Minicondaで仮想環境でjupyter notebookを利用
【事象】業務でPanadsのデータフレームでデータ処理を行っておりますが、ある列について特定の条件を満たす行についてだけ、その行の別の列の値を書き換えたいとうい状況になり困っています。簡単な例ですと、下記の様な状況です：

import pandas as pd 
df = pd.DataFrame({
   'A': [1,2,3,-1,2,-2,5],
   'B': [3,3,6,-5,2,2,-1]
   })
display(df.head(10))

ここで例えば「列Aの値が負の時は、その行の列Bの値を0にする」というようなことができれば良いのですが。。。

df['B'].loc[df.query( 'A< 0'),:]=0
display(df.head(10))

としますとエラーになりました。ValueError: Buffer has wrong number of dimensions (expected 1, got 2)

また

df['B'].where(df['A'] < 0, 0,inplace=True)
display(df.head(10))

としますとAの列も0に書き換わってしまい、、、またなぜか、A列が正の箇所が0に変わってしまっているようです。。
対策がございましたらご教示下さい。

Answer 1

こんな感じでどうでしょうか

import pandas as pd 
df = pd.DataFrame({
   'A': [1,2,3,-1,2,-2,5],
   'B': [3,3,6,-5,2,2,-1]
   })
print(df.head(10))
"""
   A  B
0  1  3
1  2  3
2  3  6
3 -1 -5
4  2  2
5 -2  2
6  5 -1
"""

#df['B'] = df.apply(lambda x: 0 if x.A < 0 else x.B, axis=1)
df.loc[df['A'] < 0,'B'] = 0
print(df)
"""
   A  B
0  1  3
1  2  3
2  3  6
3 -1  0
4  2  2
5 -2  0
6  5 -1
"""

Comments

[_hh]

できました！大変有難うございます。