Как найти сегментные индексы вектора в MATLAB?

Question

Предположим, у меня есть вектор, который содержит person_ids так:

x = [1,1,1,2,2,3,3,3,3,4] 

поэтому первые 3 элементы для человека 1. Как я могу найти векторные индексы, которые начинаются новый человек?

т.е. моя функция должна возвращать

f(x) = [1,4,6,10]; 

есть способ MATLAB для этого?

Answer 1

Предполагая, что идентификаторы персоны сортируются, и идентификаторы не могут быть отрицательными, вы можете сделать это с помощью

>> x = [1,1,1,2,2,3,3,3,3,4]; 
>> find(diff([-1 x])) 
ans = 

    1 4 6 10 

Answer 2

ответ преторианской является довольно хорошо, но если вы хотите другой альтернативы, вы можете использовать второй выход unique. Второй вывод определяет местоположения первого значения для конкретной входной последовательности.

Также предполагая, что идентификаторы человека сортируются, вы можете просто сделать это:

[~,y,~] = unique(x) 

y = 

1 
4 
6 
10 

---

Minor Примечание

Если вы посмотрите на unique documentation до версии R2013a, она используется для поиска местоположение последнее уникальное значение для входной последовательности в вашем массиве. Теперь это изменилось на первое место по сравнению с R2013a ... которое есть версия, которую я имею. Таким образом, если у вас есть версия MATLAB, которая находится перед R2013a, вам нужно добавить флаг 'first' в качестве второго параметра к уникальному. Поэтому:

[~,y,~] = unique(x, 'first'); 

Если у вас есть R2013a или выше, то вам не нужно беспокоиться об этом ... но это что-то вспомнить, должно у вас есть что-то более старый.