numpy - How to find the set of indices where two vectors have equal elements in Python -
i have 2 vectors in python: predictions , labels. find out set of indices these 2 vectors have equal elements. example, lets vectors are:
predictions = [4, 2, 5, 8, 3, 4, 2, 2] labels = [4, 3, 4, 8, 2, 2, 1, 2] so set of indices 2 vectors have equal elements be:
indices = [0, 3, 7] how can in python? without using for-loops etc. there built-in function example in numpy?
thank help!
this 1 way of doing numpy:
np.where(np.equal(predictions, labels)) which equivalent to:
np.equal(predictions, labels).nonzero() it return single element tuple though, actual array, add [0] in:
np.equal(predictions, labels).nonzero()[0]
Comments
Post a Comment