python - 在另一个列表中编码列表元素存在的最有效方法

Question

假设我有一个像这样的排序主列表:main = ["a", "b", "cd", "e"]

还有一些其他列表，例如 l1 = ["a", "cd"] 和 l2 = ["b", "cd"]。还保证了 l1 和 l2 等单个列表的所有元素也属于 main。 并非所有单个列表的长度都必须相同，但所有列表的长度都小于或等于 main。

现在，对于每个单独的列表，我想创建一个向量来编码 main 的哪些元素出现在单独的列表中。例如对于 l1，我们将得到一个向量(numpy 数组)[1,0,1,0]，因为第一个和第三个元素(我可以说“第一个”和"third"因为 main 是 main 的排序)。类似地，对于 l2，我们将得到 [0,1,1,0]。

执行此操作的有效方法是什么？我不想要天真的蛮力方法，因为那会非常慢。是否涉及任何中间 numpy 转换/操作？很乐意就此获得帮助!

Answer 1

你可以只使用列表理解:

main = ["a", "b", "cd", "e"]
l1 = ["a", "cd"]

print ([item in l1 for item in main])

输出:

[True, False, True, False]

或

print ([1 if item in l1 else 0 for item in main])

输出:

[1, 0, 1, 0]