pandasである列が指定した文字列以外の場合、別の列の文字列を書き換えたい

Question

お世話になります。

以下のようなpandasがあるとします。

import pandas as pd
df_test = pd.DataFrame({
    'col1' : ['A', 'B', 'C', 'A', 'A', 'D'],
    'col2' : ['aaa', 'bbb', 'ccc', 'aaa', 'aaa', 'ddd']
    })

	col1	col2
0	A	aaa
1	B	bbb
2	C	ccc
3	A	aaa
4	A	aaa
5	D	ddd

col1がA以外の時、col1の値をcol2に上書きしたいです。
例えばcol1がBの時、そのBという値を同じ行のcol2に上書きしたいです。
以下のようにしてみましたが上書きされず、もとのままです。

df_test[df_test['col1'] != 'A']['col2'] = df_test[df_test['col1'] != 'A']['col1']

どのようにすればいいでしょうか。最終的に以下のようなdataframeを作りたいです。

	col1	col2
0	A	aaa
1	B	B
2	C	C
3	A	aaa
4	A	aaa
5	D	D

Answer 1

df.locを使って行と列を指定。

pandasで条件に応じて値を代入（where, mask） | note.nkmk.me

import pandas as pd

df = pd.DataFrame({
    'col1': ['A', 'B', 'C', 'A', 'A', 'D'],
    'col2': ['aaa', 'bbb', 'ccc', 'aaa', 'aaa', 'ddd']
})

df.loc[df['col1'] != 'A', 'col2'] = df['col1']
print(df)

  col1 col2
0    A  aaa
1    B    B
2    C    C
3    A  aaa
4    A  aaa
5    D    D

Comments

[ggg3]

理解できました。ありがとうございました。