Pandas - заменить значения int в списке строковыми метками

Question

У меня есть фрейм данных со столбцом, который включает список идентификаторов. Я создал словарь, который связывает IDS со строковым значением. Я хотел бы заменить значения int строкой.

Name       Label_ID  
Project1   [1]
Project2   [3,5]

label_map={1:'blue', 3:'green',5:'large'}

Я хочу получить вывод:

Name       Label_ID  
Project1   [blue]
Project2   [green, large]

Я пытался использовать:

test['Tags']=test['Label_ID'].map(label_map)

Этот код выполняется, но в результате 'Теги колонка просто NaNs. Любое руководство будет оценено.

Answer 1

IIU C, вы можете взорвать свой df и затем отобразить значения, затем повторно применить groupby вдоль индекса, чтобы получить ваши значения.

обратите внимание, что ваш список может быть строкой, поэтому я добавил буквальный метод eval из библиотеки ast.

причина, по которой карта здесь не сработает, заключается в том, что вы применяете ее в виде списка, когда вам нужно применить его последовательно или поэлементно с помощью apply.

from ast import literal_eval


df["Label_Map"] = (
    df["Label_ID"]
    .apply(literal_eval) # remove if not needed.
    .explode()
    .map(label_map)
    .groupby(level=0)
    .agg(list)
)

print(df)

       Name Label_ID       Label_Map
0  Project1      [1]          [blue]
1  Project2    [3,5]  [green, large]