Find index where elements change value numpy

Question

Suppose I have

>>> v array([1, 1, 1, 1, 1, 2, 2, 2, 3, 4, 3, 4, 3, 4, 3, 4, 5, 5, 5]) 

Is there an efficient numpy way to find each index where the value changes? For instance, I would want some result like,

>>> index_of_changed_values(v) [0, 5, 8, 9, 10, 11, 12, 13, 14, 15, 16] 

If this is not possible with some numpy routine, what is a fast way to do it in python? It would also be useful to me to be referred to some good numpy tutorials since I am a numpy beginner.

Answer 1

You can get this functionality in numpy by comparing each element with it's neighbor;

v[:-1] != v[1:]   array([False, False, False, False,  True, False, False,  True,  True,     True,  True,  True,  True,  True,  True,  True, False, False], dtype=bool) 

to get the indices you use the "where" function

np.where(v[:-1] != v[1:])[0]  array([ 4,  7,  8,  9, 10, 11, 12, 13, 14, 15]) 

From here you can prepend the first element and add a one to get to the same indexing scheme you have in your question.