python - 如何在Python中轻松将标签集转换为所属元组("multi-hot"编码)？

Question

我有一组标签tags和所有可能标签taglist的有序列表。现在我想将一组标签转换为“多热”编码，即获取一个与 taglist 长度相同的列表或元组，其中在所属标签所在的位置有 1，在其他位置有 0地点。

目前我做的是直接:

        multihot = []
        for i in range(len(taglist)):
            tag = taglist[i]
            if tag in tags:
                multihot.append(1)
            else:
                multihot.append(0)

可以写一行吗？

Answer 1

multihot = [(tag in tags) for tag in tag_list] 

multihot 现在包含 True 或 False 条目。在Python中，True的行为与1完全相同，False的行为与0完全相同；所以这相当于你的解决方案。