【Python】既存のデータフレームの一部の値を別のデータフレームの値で上書きする

Question

前提・実現したいこと

素人質問で恐縮です。
Pythonのデータ処理にて、データフレーム1(df1)とデータフレーム2(df2)を、共通のIDデータをキーにして結合させる。その際、df1の'count'列の値をdf2の'count'列の値で上書きした状態で結合させたいです。ただ、df1とdf2のデータ数は異なっており、df1の'ID'列 3 における'count'の値を、df2の'ID'列 3における'count'の値で上書きした状態で結合するというイメージです。

以下実現したいイメージです。

df1
ID count
1 5
2 4
3 3
4 2
5 1

df2
ID count
3 5
5 2

df1とdf2を結合したdf1_new
ID count
1 5
2 4
3 5
4 2
5 2

試したこと

pandasのmerge関数を用いれば出来るのではないかと考え、結合の条件を変えてみましたが、どれも上記のイメージにはなりませんでした。

Answer 1

こんにちは。

こんな感じでいかがでしょうか？

import pandas as pd

df_ab = pd.DataFrame({'ID': [1, 2, 3, 4, 5], 'count': [5, 4, 3, 2, 1]})
df_ac = pd.DataFrame({'ID': [3, 5], 'count': [5, 2]})

# https://note.nkmk.me/python-pandas-merge-join/
df_merge = pd.merge(df_ab, df_ac, on='ID', how='outer')

# https://note.nkmk.me/python-pandas-nan-dropna-fillna/
print(
    pd.DataFrame({
        'ID': df_merge['ID'],
        'count': df_merge['count_y'].fillna(df_merge['count_x']),
    })
)

Comments

[tsumiki_kuzushi]

ご回答ありがとうございます！別回答者から頂いたアプローチと異なりますが、こちらのアプローチでも問題解決できました！

[退会済みユーザー]

お、うまくいってよかったです。
また何か困ったことがありましたら、気軽にご連絡ください〜

Enjoy!!

Answer 2

こんな感じです。

>>> print(df1)
   ID  count
0   1      5
1   2      4
2   3      3
3   4      2
4   5      1
>>> print(df2)
   ID  count
0   3      5
1   5      2
>>> df1_temp = df1.set_index('ID')
>>> df1_temp.update(df2.set_index('ID'))
>>> df1_new = df1_temp.astype(int).reset_index()
>>> print(df1_new)
   ID  count
0   1      5
1   2      4
2   3      5
3   4      2
4   5      2

Comments

[tsumiki_kuzushi]

解決いたしました！ありがとうございました！