python - 使用映射转换 Pandas 数据框

Question

我有一个 Pandas 数据框。有客户信息的df1:

Customer_Name    Demand
John               100
Mike               200
...

还有一个字典，里面有客户名字和客户代码的映射

Customer_Name    Customer_Code
John               1
Mike               2
...

我想提出一个像 Df1 这样的新数据框，但使用客户代码而不是名称:

Customer_Code    Demand
    1              100
    2               200
    ...

为此，我使用了以下代码:

df3=data.replace({"customer_code": mapp})
Raw=data_m[['Demand','customer_code]]

它给了我正确的结果，但是速度很慢。我想知道是否有更有效的方式进行这种映射和转换？

Answer 1

merge 应该就可以了。

df = df1.merge(df2)
df
  Customer_Name  Demand  Customer_Code
0          John     100              1
1          Mike     200              2

如果您想删除第一列，请调用 df.drop('Customer_Name', 1):

df.drop('Customer_Name', 1)
   Demand  Customer_Code
0     100              1
1     200              2

或者，列索引:

df[['Customer_Code', 'Demand']]
   Customer_Code  Demand
0              1     100
1              2     200

---

或者，您可以使用 df.map

df1['Customer_Code'] = df1.Customer_Name.map(\
             df2.set_index('Customer_Name').Customer_Code)

df1
  Customer_Name  Demand  Customer_Code
0          John     100              1
1          Mike     200              2