[python] 新規列に条件に合うデータを追加したい

Question

実現したいこと

下記のようなdataframeで条件にあうデータを追加したいです
（都道府県列を判定してエリア列に該当のエリアを追加）

|ID|日付|種別|都道府県|エリア| ←エリアを新規列
|1111|2022/1/1|新規|東京　|関東|
|2222|2022/1/1|新規|神奈川|関東|
|3333|2022/1/2|変更|大阪　|大阪|

該当のソースコード

import pandas as pd
import numpy as np

df_test = pd.DataFrame({'ID': [1111, 2222,  3333],
                                     '日付': [2022/1/1, 2022/1/1, 2022/1/2],                                      
                                     '種別': ['新規', '新規', '変更'],
                                      都道府県': ['東京', '神奈川', '大阪']})

①の方法
adict = pd.DataFrame({'都道府県': ['東京','神奈川','大阪'], 
                                   'エリア': ['関東','関東','大阪']})
df = pd.concat([df_test, adict])
df.assign(area = lambda x:np.where(x["都道府県"] == df['都道府県'], df['エリア'],0))

--------------------------------------------------------
②の方法
bdict = ({'都道府県': ['東京','神奈川','大阪'],  'エリア': ['関東','関東','大阪']})
for k, v in bdict.items():
    if (row.'都道府県' == k):
        return v

試したこと

①、②の方法を考えましたが実現が上手く出来ず、ご指摘を頂きたいです。

Answer 1

キーを都道府県でmergeすればよいかと思います。

import pandas as pd
import numpy as np

df_test = pd.DataFrame({'ID': [1111, 2222,  3333],
                                     '日付': [2022/1/1, 2022/1/1, 2022/1/2],                                      
                                     '種別': ['新規', '新規', '変更'],
                                      '都道府県': ['東京', '神奈川', '大阪']})

# ①の方法
adict = pd.DataFrame({'都道府県': ['東京','神奈川','大阪'], 
                                   'エリア': ['関東','関東','大阪']})

df = df_test.merge(adict, on='都道府県')
print(df)
"""
     ID      日付  種別 都道府県 エリア
0  1111  2022.0  新規   東京  関東
1  2222  2022.0  新規  神奈川  関東
2  3333  1011.0  変更   大阪  大阪
"""

Comments

[melian]

merge のデフォルトは inner join なので how='left' を付けておくと良いかと。

[mango55]

早々のお返事ありがとうございます。実現出来ました。
今回は先にご回答頂いた方をベストアンサーとさせて頂き、すみません。様々な実現方法を教えて頂き、勉強になります。
いつもありがとうございます。

Answer 2

import pandas as pd

df_test = pd.DataFrame({'ID': [1111, 2222,  3333],
                        '日付': ['2022/1/1', '2022/1/1', '2022/1/2'],
                        '種別': ['新規', '新規', '変更'],
                        '都道府県': ['東京', '神奈川', '大阪']})
adict = pd.DataFrame({'都道府県': ['東京','神奈川','大阪'],
                      'エリア': ['関東','関東','大阪']})

df_test['エリア'] = df_test['都道府県'].map(adict.set_index('都道府県')['エリア'])
print(df_test)

ID	日付	種別	都道府県	エリア
1111	2022/1/1	新規	東京	関東
2222	2022/1/1	新規	神奈川	関東
3333	2022/1/2	変更	大阪	大阪

Comments

[mango55]

早々のお返事ありがとうございます。実現出来ました。
勉強になりました。いつもありがとうございます。