python——使用map快速匹配字典的

# -*- coding:utf-8 -*-
import pandas as pd
import warnings

warnings.filterwarnings('ignore')

Type_Dict = {
    u"小红": 1,
    u"小明": 2,
    u"小刚": 3,
    u"小绿": 4,
    u"什么鬼": 5
}

list = [u'小红', u'小绿', u'小刚', u'小明', u'看看']

a = pd.DataFrame(list)

a.columns = ['name']

a['type'] = a['name'].map(Type_Dict)

场景很简单，如上面的代码，一个字典type_Dict，有五个键值对，随后我们新建了一个list并转换为Dataframe，列名为name。

a['type'] = a['name'].map(Type_Dict)，Dataframe变成如下：

是不是很方便？