pandas 在python中将标签列表转换为每个ID的标签数组

rpppsulh 于 2023-04-04 发布在 Python

关注(0)|答案(1)|浏览(113)

我有一个包含文本和标签的数据框。每个文本都有多行标签。

dummy_df = pd.DataFrame([['Text1','label1'], ['Text1', 'label2']], columns=["TEXT", "LABELS"])

我想有以下应用MultiLabelBinarizer（）函数。

TEXT | LABEL
Text1| [[label1,label2]]

Reference 1 Reference 2

pandas

来源：https://stackoverflow.com/questions/75917657/translate-list-of-labels-into-array-of-labels-per-id-in-python

1条答案

按热度按时间

xlpyo6sf1#

如果需要嵌套列表，请使用GroupBy.agg中的lambda函数：

df = dummy_df.groupby('TEXT')['LABELS'].agg(lambda x: [x.tolist()]).reset_index()
print (df)
    TEXT              LABELS
0  Text1  [[label1, label2]]

非嵌套列表：

df1 = dummy_df.groupby('TEXT')['LABELS'].agg(list).reset_index()
print (df1)
    TEXT            LABELS
0  Text1  [label1, label2]

赞(0）回复(0）举报 2023-04-04

我来回答

pandas 在python中将标签列表转换为每个ID的标签数组

1条答案

相关问题

热门标签

最新问答