Панды dataframe - добавление столбца из Словаре

Question

Я имею

df = pd.DataFrame.from_dict({'col1':['A','B', 'B', 'A']}) 
    col1 
0 A 
1 B 
2 B 
3 A 


    other_dict = {'A':1, 'B':0} 

Я хочу добавить столбец в ДФ, так что это выглядит следующим образом:

col1 col2 
0 A  1 
1 B  0 
2 B  0 
3 A  1 

Answer 1

Вы также можете использовать map:

In [3]: 
df['col2'] = df['col1'].map(other_dict) 
df 

Out[3]: 
    col1 col2 
0 A  1 
1 B  0 
2 B  0 
3 A  1 

Answer 2

Одним из вариантов является использование применяются:

In [11]: df["col1"].apply(other_dict.get) 
Out[11]: 
0 1 
1 0 
2 0 
3 1 
Name: col1, dtype: int64 

затем присвоить его колонке:

df["col2"] = df["col1"].apply(other_dict.get) 

---

Другой, который может быть более эффективным (если у вас есть большие группы) заключается в использовании преобразования:

In [21]: g = df.groupby("col1") 

In [22]: g["col1"].transform(lambda x: other_dict[x.name]) 
Out[22]: 
0 1 
1 0 
2 0 
3 1 
Name: col1, dtype: object 

---

Стоит также ссылки на categorical section of the docs.