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变成如下：

是不是很方便？