別のデータフレームから条件に合うデータを追加したい

Question

二つのデータフレームがあったとします。

df1
|index|A|B|C|D|E|F|
|:--|:--:|--:|
|0|111|NaN|666|Nan|0.1|Nan|
|1|222|NaN|777|Nan|0.2|Nan|
|2|333|NaN|888|Nan|0.3|Nan|

df2
|index|aa|bb|
|:--|:--:|--:|
|0|111|あああ|
|1|222|いいい|
|2|333|ううう|
|2|666|えええ|
|2|777|おおお|
|2|888|かかか|

df1のNanの場所にデータを入れるのですが、
df1のA列とdf2のaa列が同じ時に、df2のbb列のデータを入れるのが目標です。

想定結果
|index|A|B|C|D|E|F|
|:--|:--:|--:|
|0|111|あああ|666|えええ|0.1|Nan|
|1|222|いいい|777|おおお|0.2|Nan|
|2|333|ううう|888|かかか|0.3|Nan|

このような結果になるようにするには、pythonでどのように記載すればよいのでしょうか。
また、このようなことが可能でしょうか。

Answer 1

pandas.mergeを使うといいかと思います。質問に記載されたケースだと以下のようなコードになります。

import pandas as pd
pd.merge(df1,df2,left_on='A',right_on='aa',how='left')

上記によってdf1のA列とdf2のaa列の値が同じ場合に値が入ったbb列がdf1に付与されます。なお、同じではnない場合はnanが入りますので、必要に応じてfillnaを使って補完してください。最終的にB列に値を入れるのであれば、df1['B']=df1['bb']としてください。

Comments

[sarupip]

ありがとうございました。
列全体で一気にできそうなので、行数が多いときに重宝しそうです。
ありがとうございました。
行が多くてもできそうなので、ベストアンサーにさせていただきます。

Answer 2

pandas.DataFrameで定義されたデータフレームの場合、df2.loc[df2.aa==111,"bb"]のような書き方で、df2から必要なデータを表引きできます。

これをdf1の適切な場所に代入していけば目標は達成できるでしょう。

Comments

[sarupip]

ありがとうございます。
このようなこのような書き方でできました。ありがとうございました。

Answer 3

df2から変換用のSeriesデータを生成し、df1の'A'列と'C'列をmap()にて変換するという手もあります。

import pandas as pd
import numpy as np

df1 = pd.DataFrame(
    [[111,np.nan,666,np.nan,0.1,np.nan],
     [222,np.nan,777,np.nan,0.2,np.nan],
     [333,np.nan,888,np.nan,0.3,np.nan]],
    columns=['A','B','C','D','E','F'])
df2 = pd.DataFrame(
    [[111,'あああ'],
     [222,'いいい'],
     [333,'ううう'],
     [444,'えええ'],
     [555,'おおお'],
     [666,'かかか']],
     columns=['aa','bb'])


df1['B'] = df1['A'].map(df2.set_index('aa')['bb'])
df1['D'] = df1['C'].map(df2.set_index('aa')['bb'])
print(df1)

Comments

[sarupip]

この書き方もすっきりして見やすいなと思いました。
余力があるときに試してみようと思います。
お勉強の役に立ちます。ありがとうございました。