gpt4 book ai didi

python - 确定两个 numpy 数组在 Python 中相交的参数

转载 作者:太空狗 更新时间:2023-10-29 23:59:31 25 4
gpt4 key购买 nike

我有两个数组,比如说:

a, b = np.array([13., 14., 15., 32., 33.]), np.array([15., 16., 17., 33., 34., 47.])

我需要找到 a 中不存在于 b 中的所有元素的索引。在上面的示例中,结果将是:

[0, 1, 3]

因为a[0]、a[1]和a[3]是13.、14.和32.,它们不存在于b中。请注意,我不想知道 13.、14. 和 32. 的实际值。(在这种情况下,我可以使用 set(a).difference(set(b)))。我真的只对指数感兴趣。

如果可能,答案应该是“向量化的”,即不使用 for 循环。

最佳答案

你可以使用 np.in1d :

>>> np.arange(a.shape[0])[~np.in1d(a,b)].tolist()
[0, 1, 3]

关于python - 确定两个 numpy 数组在 Python 中相交的参数,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/18486001/

25 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com