我在Python中有两个向量：Predictions和Labels。我想做的是找出这两个向量具有相等元素的索引集。例如，假设向量为：
Predictions = [4, 2, 5, 8, 3, 4, 2, 2]
Labels = [4, 3, 4, 8, 2, 2, 1, 2]
因此，两个向量具有相等元素的索引集为：
Indices = [0, 3, 7]
如何在Python中获得此信息？不使用for循环等。是否有内置函数numpy？
感谢您的任何帮助！
这是使用numpy的一种方法：
np.where(np.equal(Predictions, Labels))
等效于：
np.equal(Predictions, Labels).nonzero()
但是它将返回单个元素元组，因此要获得实际的数组，请添加[0]如下内容：
np.equal(Predictions, Labels).nonzero()[0]
本文收集自互联网，转载请注明来源。
如有侵权，请联系 [email protected] 删除。
我来说两句