Экстрагирования чередование из вектора R

Question

Я имею выглядя данные, как следующее:

A= c(0,0,0,-1,0,0,0,1,1,1,0,0,-1,0,0,-1,-1,1,1,1,-1,0,0,0,-1,0,0,-1,-1,1,1,0,0,0,0,1,-1) 

Цель состоит в том, чтобы извлечь чередующиеся -1S и 1s. Я хочу создать функцию, в которой входной вектор содержит 0,1 и -1. Выход идеально выделяет все 0s и чередует -1s и 1s.

Например, требуемый выход для приведенного выше примера:

B= c(0,0,0,-1,0,0,0,1,0,0,0,0,-1,0,0,0,0,1,0,0,-1,0,0,0,0,0,0,0,0,1,0,0,0,0,0,0,-1) 

Два 1s в 9-м и 10-м месте в А оказалось равным 0, потому что мы только сохранить первое 1 или -1 появляться. По этой причине -1s в 16-м и 17-м местоположениях A также поворачиваются на 0.

У кого-нибудь есть хорошая идея для создания такой функции?

Answer 1

Определение позиций ненулевых значений:

w = which(A != 0) 

Для каждого прогона аналогичных значений, в A[w], занимать позицию первый:

library(data.table) 
wkeep = tapply(w, rleid(A[w]), FUN = function(x) x[1]) 

Установить все остальные значения до нуля:

# following @alexis_laz's approach 
B = numeric(length(A)) 
B[ wkeep ] = A[ wkeep ] 

Таким образом, вам не нужно делать сравнения в цикле, в котором R медленнее, я думаю.

---

rleid происходит из данных.table. С базой R, вы можете сделать wkeep с предложением @ alexis_laz в:

wkeep = w[c(TRUE, A[w][-1L] != A[w][-length(w)])] 

Или написать свой собственный rleid, как и в Josh's answer.

Answer 2

Вы должны сдвинуть весь массив и переменную флага, которую вы проверяете, если ранее вы обнаружили 1 или -1. это может быть возможным псевдокод алгоритма:

while i < length(a): 

    if flag == 1 && a[i]=-1: 
     b[i]=a[i]; 
     flag = -1; 
    else if flag == -1 && a[i] = 1: 
     b[i]=a[i]; 
     flag = 1; 
    else: 
     b[i]=0; 
    i++; 
}//end of while 

Answer 3

Это действительно просто Конкретизация псевдо-кода GWarius в. (Я уже имел структуру, но логик, которая была неисправной.)

last1 <- -A[which(A != 0)[1] ] # The opposite of the first non-zero item 
for (i in seq_along(A)){ 
      if(last1==1 && A[i]==-1 ){ last1 <- -1 
      } else {if (last1 == -1 && A[i] == 1) { last1 <- 1 
       } else {A[i] <- 0}} } 
A 
[1] 0 0 0 -1 0 0 0 1 0 0 0 0 -1 0 0 0 0 1 0 0 -1 0 0 
[24] 0 0 0 0 0 0 1 0 0 0 0 0 0 -1 

> identical(A, B) 
[1] TRUE