Pandasで表から要素を検索して新たな列を付け加えたい。

Question

例えば、以下のようなデータと検索表があったとします。

#データ
data = pd.DataFrame(["c","a","b","a","a"],columns=["データ"])
#検索表
search = pd.DataFrame([["a",1],["b",2],["c",3]],columns=["検索表","要素"])

検索表の「検索表」列を参考にしてデータに「要素」列を追加したいと考えています。結果は以下の画像のようになってほしいです。

pd.DataFrame([["c",3],["a",1],["b",2],["a",1],["a",1]],columns=["データ","要素"])

分かりにくい説明となりましたが、ご回答よろしくお願いします。

Answer 1

pd.merge(data, search.rename(columns={"検索表":"データ"}), how="left")

または

pd.merge(data, search, how="left", left_on="データ", right_on="検索表")[["データ", "要素"]]

Answer 2

pd.mergeを使いましょう

>>> import pandas as pd
>>> data = pd.DataFrame(["c","a","b","a","a"],columns=["データ"])
>>> search = pd.DataFrame([["a",1],["b",2],["c",3]],columns=["検索表","要素"])
>>> df = pd.merge(data, search, left_on="データ", right_on="検索表", how="left").drop("検索表", axis=1)
>>> df
  データ  要素
0   c   3
1   a   1
2   b   2
3   a   1
4   a   1

Answer 3

この程度の処理（失礼）であれば、pd.merge()よりもSeries.map() や Series.replace()の処理で十分ではないでしょうか

data['要素'] = data['データ'].map(search.set_index('検索表')['要素'])

https://pandas.pydata.org/pandas-docs/stable/reference/api/pandas.Series.map.html

---

上記のコードでは、ちょっと分かりにくいかもしれないので一応説明すると

conv = search.set_index('検索表')['要素']
#検索表
#a    1
#b    2
#c    3
#Name: 要素, dtype: int64

のように、indexが'検索表、データ列が要素のSeriesデータ（変換テーブル）を生成しておいて

data['要素'] = data['データ'].map(conv)

のように dataのデータ列に上記の変換テーブルを適用して、その結果を要素列に入れているだけです。