how to create a vector of consecutive numbers that skip over certain elements

I have a vector K whose elements are either 1 or 0 and whose length may change depending on other inputs. I want code that will create a vector R with equal number of elements as K that equals consecutive numbers (1, 2, 3...) except where K=1. Where K=1 I want R to equal 0.
For instance
K might be:
K=[0; 0; 1; 0; 0; 1; 1; 0; 0; 1]
then I would like syntax that creates R as:
R=[1; 2; 0; 3; 4; 0; 0; 5; 6; 0]
Thanks for the help

1 comentario

You can do like that
K = [0; 0; 1; 0; 0; 1; 1; 0; 0; 1];
M = ~K;
L = (1:length(K))')
R = (immultiply(L, M))';

Iniciar sesión para comentar.

 Respuesta aceptada

Another approach:
K=[0; 0; 1; 0; 0; 1; 1; 0; 0; 1]
R = cumsum(K==0)
R(K==1) = 0

Más respuestas (2)

Walter Roberson
Walter Roberson el 21 de Oct. de 2013
Editada: Walter Roberson el 21 de Oct. de 2013
R = (1:length(K)).';
R(K==1) = 0;
Or
R = (1:length(K)).' .* (1-K);

2 comentarios

Thanks. I think this is close, but not exactly what I asked for. Both code snippets you posted would give me the following:
K=[0; 0; 1; 0; 0; 1; 1; 0; 0; 1]
then I would like syntax that creates R as:
R=[1; 2; 0; 4; 5; 0; 0; 8; 9; 0]
What I want is something that will recreate this:
K=[0; 0; 1; 0; 0; 1; 1; 0; 0; 1]
then I would like syntax that creates R as:
R=[1; 2; 0; 3; 4; 0; 0; 5; 6; 0]
So, the consecutive numbering "skips" over the zeros.
n = nnz(K);
R = zeros(size(K,1),1);
R( K ~= 0 ) = 1:n;

Iniciar sesión para comentar.

Categorías

Más información sobre Simulink en Centro de ayuda y File Exchange.

Productos

Etiquetas

Preguntada:

el 21 de Oct. de 2013

Respondida:

el 21 de Oct. de 2013

Community Treasure Hunt

Find the treasures in MATLAB Central and discover how the community can help you!

Start Hunting!

Translated by