dataframeから特定の条件を満たす行のみに対して値を加算する方法

Question

以下のデータフレームに対して、addFlag列がyesの行に対してのみ'forAdd`列の値を
x,y列に加算したいのですがうまく行きません。
下記のようにトライして見たのですが、どこを修正すれば良いかご教示いただけないでしょうか。

import pandas as pd 

df = pd.DataFrame({'x': [1, 2, 3, 4, 5],
                   'y': [6, 7, 8, 9, 10],
                   'addFlag': ['yes', 'no', 'yes', 'no', 'yes'],
                   'forAdd':[100,100,100,100,100]
                   })

condition = df['addFlag'].str.contains('yes')
df.loc[condition,['x','y']] += df.loc[condition,'forAdd']

df

#実行前のdataframe
	x    y	addFlag	forAdd
0	1	6	yes	100
1	2	7	no	100
2	3	8	yes	100
3	4	9	no	100
4	5	10	yes	100


#実行結果
	x       　y	addFlag forAdd
0	NaN	NaN	yes	100
1	2.0	7.0	no	100
2	NaN	NaN	yes	100
3	4.0	9.0	no	100
4	NaN	NaN	yes	100

達成したいこと：NaNの部分は100が足された値になってほしい

Answer 1

データフレームの+演算子ではカラム方向に足し算しようとするのでおかしくなります。

addをaxis=0を指定して、行方向に足してやればいいです。

df.loc[condition, ['x', 'y']] = df[['x', 'y']].add(df['forAdd'], axis=0)

他の回答者さんの回答のように、列ごとに書く方が綺麗かもしれません。

Comments

[lorree]

ありがとうございます。データフレームでの演算子の挙動は知らなかったので勉強になりました。

Answer 2

こういう感じです。

>>> import pandas as pd
>>>
>>> df = pd.DataFrame({'x': [1, 2, 3, 4, 5],
...                    'y': [6, 7, 8, 9, 10],
...                    'addFlag': ['yes', 'no', 'yes', 'no', 'yes'],
...                    'forAdd':[100,100,100,100,100]
...                    })
>>>
>>> condition = df['addFlag'].str.contains('yes')
>>> df.loc[condition,'x'] = df['x'] + df['forAdd']
>>> df.loc[condition,'y'] = df['y'] + df['forAdd']
>>> print(df)
     x    y addFlag  forAdd
0  101  106     yes     100
1    2    7      no     100
2  103  108     yes     100
3    4    9      no     100
4  105  110     yes     100

Comments

[lorree]

ありがとうございます！

df.loc[condition,'x'] = df['x'] + df['forAdd']

は左辺と右辺でSeriesの長さが異なるため一見エラーになりそうに思えるのですが、どのように理解するのがいいのでしょうか...？

Answer 3

すなおに２行に分けてセットすればよいかと思います。

import pandas as pd 

df = pd.DataFrame({'x': [1, 2, 3, 4, 5],
                   'y': [6, 7, 8, 9, 10],
                   'addFlag': ['yes', 'no', 'yes', 'no', 'yes'],
                   'forAdd':[100,100,100,100,100]
                   })

condition = df['addFlag'].str.contains('yes')
add = df[condition]['forAdd']

df.loc[condition,'x'] += add
df.loc[condition,'y'] += add
print(df)
"""
     x    y addFlag  forAdd
0  101  106     yes     100
1    2    7      no     100
2  103  108     yes     100
3    4    9      no     100
4  105  110     yes     100
"""

Comments

[lorree]

ありがとうございます。add = df[condition]['forAdd']としておけば足し算が可能になるのですね。大変勉強になりました。