DataFrame A のTypeコラムに、 DataFrame B の相当する情報を当てはめたい

Question

<やりたいこと>
DataFrame A　のTypeコラムに、
DataFrame B　の相当する情報を当てはめてA'を作りたいのですが、
どのように結合すればいいのかわかりません。
DataFrame Aの１を色に置き換えて0をDropするのかな。。ということで試してみたのですが、
なかなかうまく進みません。

手順だけでもいいのでお教えいただけませんでしょうか。
何卒宜しくお願い致します。

DataFrame A
|Name|Red|yellow|Green|
|:--|:--:|--:|
|A|0|1|0|
|B|0|0|1|
|C|0|1|0|
|D|1|0|0|
|E|0|1|0|

Data Frame B
|Number|Place|Color|Type|
|:--|:--:|--:|
|1|Tokyo|A|Car|
|2|Osaka|B|Train|
|3|Narita|C|Ship|
|4|Chiba|D|Walk|
|5|Tokyo|E|Car|

を結合して
Dataframe A'を作りたい
|Number|Place|Color|Type|
|:--|:--:|--:|
|1|Tokyo|Yellow|Car|
|2|Osaka|Green|Train|
|3|Narita|Yellow|Ship|
|4|Chiba|Red|Walk|
|5|Tokyo|Yellow|Car|

宜しくお願い致します。

Answer 1

B['Color'] = B['Color'].map(A.set_index('Name').idxmax(axis=1))

でどうでしょうか。

一応、動作サンプル

import pandas as pd

A = pd.DataFrame({
    'Name':['A','B','C','D','E'],
    'Red':[0,0,0,1,0],
    'Yellow':[1,0,1,0,1],
    'Green':[0,1,0,0,0]
})

B = pd.DataFrame({
    'Number':[1,2,3,4,5],
    'Place':['Tokyo','Osaka','Narita','Chiba','Tokyo'],
    'Color':['A','B','C','D','E'],
    'Type':['Car','Train','Ship','Walk','Car']
})
B = B.loc[:,['Number', 'Place', 'Color', 'Type']]



B['Color'] = B['Color'].map(A.set_index('Name').idxmax(axis=1))
print(B)
#   Number   Place   Color   Type
#0       1   Tokyo  Yellow    Car
#1       2   Osaka   Green  Train
#2       3  Narita  Yellow   Ship
#3       4   Chiba     Red   Walk
#4       5   Tokyo  Yellow    Car

Comments

[trey_0329]

ご回答ありがとうございます！無事に解決いたしました