python - 使用 numpy 获取数组列表中相交元素的计数(避免循环)

Question

我有一个名为 MyFruits 的值数组，例如:

[apple, orange, banana, apple, pear]

然后我有一个数组列表，例如:

[apple, orange]
[blueberry, watermelon, pear]
[grape, orange, grape, orange]
[]
[cantaloupe]

对于列表中的每个数组，我想得到与 MyFruits 数组相交的元素数除以数组中的元素总数。所以输出将是:

2 / 2 = 1
1 / 3 = 0.66667
2 / 4 = 0.5
0 / 0 = (in this case 0)
0 / 1 = 0

本质上:

[1, 0.66667, 0.5, 0, 0]

我一直在使用 for 循环在 Python 中执行此操作，但是数据集很大而且速度非常慢。有人建议使用 numpy，但我很难理解。

Answer 1

假设您有两个列表，一个长度为 M，另一个长度为 N。如果通过直接的线性搜索来完成，则需要 O(M * N) 次字符串比较才能找到两个列表中的哪些元素。

您可以使用 Python 集合对此进行改进。将列表转换为 Python 集并使用 set intersection (&)找到他们的共同点。然后复杂度降低到 O(M + N)。