使用字典键创建新列

Question

拿这个 dataframe df ：

df = pd.DataFrame({'col_1':['some text green', 'some text blue', 'some text dog', 'some text']})

             col_1
0  some text green
1   some text blue
2    some text dog
3        some text

拿这本字典：

my_dict = {'color':['green', 'blue'],'animal':['dog']}

如果字符串包含在col_1中，我需要创建一个新列new_col搜索。 字符串在字典值中给出。 如果是这样，我需要获取字典键并将其放在新列中。 如果没有，只需放置 NaN。

结果应该是：

             col_1 new_col
0  some text green   color
1   some text blue   color
2    some text dog  animal
3        some text     NaN

Answer 1

尝试这个：

def f(s):
    for k in my_dict:
        for v in my_dict[k]:
            if v in s:
                return k

df["new_col"] = df["col_1"].apply(f)