pythonのdataframeの値を条件で他の列から同じ代入

Question

前提・実現したいこと

pythonのdataframeの値を条件で他の列から代入したい

該当のソースコード

G 背 力 体力
1 1 555 0
1 2 666 0
1 3 444 0
1 4 333 0
1 5 222 0
2 1 111 0
2 2 555 0
2 3 666 0
2 4 777 0
2 5 222 0
3 1 111 0
3 2 333 0
3 3 222 0
3 4 111 0
3 5 555 0
4 1 666 0
4 2 777 0
4 3 222 0
4 4 111 0
4 5 333 0
みたいなpythonのdataframeでGごとに同じ力を入れたいです。
背が2の力を体力に入れるみたいな。
つまり2を指定したら、こうなるように
G 背 力 体力
1 1 555 666
1 2 666 666
1 3 444 666
1 4 333 666
1 5 222 666
2 1 111 555
2 2 555 555
2 3 666 555
2 4 777 555
2 5 222 555
3 1 456 333
3 2 333 333
3 3 222 333
3 4 111 333
3 5 555 333
4 1 666 777
4 2 777 777
4 3 222 777
4 4 111 777
4 5 333 777
groupbyを使えばいいのか色々考えましたが、分かりませんでした。
Gはindexでもそうじゃなくても、
体力は既存の列でも、新たに列を作ってもいいです。

Answer 1

import pandas as pd
import io

txt = """
G,背,力,体力
1,1,555,0
1,2,666,0
1,3,444,0
1,4,333,0
1,5,222,0
2,1,111,0
2,2,555,0
2,3,666,0
2,4,777,0
2,5,222,0
3,1,111,0
3,2,333,0
3,3,222,0
3,4,111,0
3,5,555,0
4,1,666,0
4,2,777,0
4,3,222,0
4,4,111,0
4,5,333,0
"""

df = pd.read_csv(io.StringIO(txt))
# print(df)

dic = df.loc[df['背'] == 2, ['G', '力']].set_index('G')['力'].to_dict()
# print(dic)

for _, s in df.iterrows():
    s['体力'] = dic[s['G']] if s['G'] in dic else None

print(df)

    G  背    力   体力
0   1  1  555  666
1   1  2  666  666
2   1  3  444  666
3   1  4  333  666
4   1  5  222  666
5   2  1  111  555
6   2  2  555  555
7   2  3  666  555
8   2  4  777  555
9   2  5  222  555
10  3  1  111  333
11  3  2  333  333
12  3  3  222  333
13  3  4  111  333
14  3  5  555  333
15  4  1  666  777
16  4  2  777  777
17  4  3  222  777
18  4  4  111  777
19  4  5  333  777

Comments

[pythondata]

ありがとうございます。自分には複雑すぎて答え導くのが無理でした

[Daregada]

解決方法は何通りもあるので、効率を気にせず自分でわかる方法を使えばいいと思います。

[pythondata]

はい。そうですね。何行にもなっちゃうところでした