怎么通過pandas column內(nèi)容匹配字典的key來添加一列

小編給大家分享一下怎么通過pandas column內(nèi)容匹配字典的key來添加一列，相信大部分人都還不怎么了解，因此分享這篇文章給大家參考一下，希望大家閱讀完這篇文章后大有收獲，下面讓我們一起去了解一下吧！

這是知乎上的問題，原問題是這樣的：

我想根據(jù)H列的交易對(duì)方的關(guān)鍵字，生成Q列類型。比如H列第一個(gè)包含“美的“，那Q列就是"電器"。

如果一個(gè)一個(gè)添加我會(huì)，我想生成一個(gè)字典{"官方":"電器"，“網(wǎng)絡(luò)技術(shù)”:"投資"……}，通過匹配dataframe中H列內(nèi)容包含字典key，來添加一列，匹配類型。請(qǐng)問如何操作？

我來回答一下：

''總體思路是利用遍歷字典的key值，然后使用in判斷，如果true則返回value'''
# 構(gòu)造的字典
>>> my_dict ={'bo': 'A', 'jik': 'B', 'luc': 'C'}
# 構(gòu)造的dataframe
>>> df = pd.read_excel(r'D:/myExcel/1.xlsx')
>>> df
    name  score
0    bob     45
1  jiken     67
2    bob     23
3    bob     34
4   lucy     42
5   lucy     75
# 定義函數(shù)，進(jìn)行邏輯判斷
>>> def get_res(strings):
    for key, value in my_dict.items():
        if key in strings:
            return value
    return ""
# 對(duì)name的每一行運(yùn)用apply函數(shù)，獲取返回結(jié)果，并返回grade列
>>> df['grade'] = df['name'].apply(lambda x: get_res(x))
>>> df
    name  score grade
0    bob     45     A
1  jiken     67     B
2    bob     23     A
3    bob     34     A
4   lucy     42     C
5   lucy     75     C

以上是“怎么通過pandas column內(nèi)容匹配字典的key來添加一列”這篇文章的所有內(nèi)容，感謝各位的閱讀！相信大家都有了一定的了解，希望分享的內(nèi)容對(duì)大家有所幫助，如果還想學(xué)習(xí)更多知識(shí)，歡迎關(guān)注億速云行業(yè)資訊頻道！