从另一个具有字典键的列创建pandas dataframe列

Question

我有数据框和字典。 这些看起来像

import pandas as pd
df1 = pd.DataFrame({'first':['john','oliver','sarah']})
df1_map = {'john': 'anderson', 'oliver': 'smith', 'sarah' : 'shively'}
print (df1)
print (df1_map)

    first
0    john
1  oliver
2   sarah
{'oliver': 'smith', 'sarah': 'shively', 'john': 'anderson'}

df1 ['first']的值表示字典的键值。

我想在数据帧中添加第二列df1 ['second']，以便保持dict关系以获得以下数据帧，

    first    last
0    john  anderson
1  oliver     smith
2   sarah   shively

现在，我可以像这样遍历数据框的值，

df1['last'] = [ df1_map[i] for i in list(df1['first'])]

我想知道熊猫是否支持矢量化的实现/功能，而该实现/功能可以做到这一点而无需遍历df的行

Answer 1

您可以使用以下命令将字典值直接映射到键：

df1['last'] = df1['first'].map(df1_map)

结果是：

Out[6]: 
    first      last
0    john  anderson
1  oliver     smith
2   sarah   shively