행렬 요소 조작

Question

b = c(1,1,2,2,3,3,4,4,1) 
c = c(10,10,20,20,30,30,40,40,5) 
a <- NULL 
a <- matrix(c(b,c), ncol=2) 

이 행렬의 첫 번째 열에서 숫자를 비교하고 첫 번째 숫자가 열의 두 번째 연속 번호와 같은지 비교합니다 (이 경우 1 = 1 인 경우). , 등) 두 번째 열에 함께 해당 숫자를 추가하려면 (10 + 10 = 20 등) 한 값일 것이므로이 출력을 별도의 벡터에 저장하고 싶습니다 . 다음과 같이

내가 찾고 있어요 매트릭스의 출력은 다음과 같습니다

 [,1] [,2] [,3] 
[1,] 1 10 20 
[2,] 1 10 40 
[3,] 2 20 62 
[4,] 2 20 85 
[5,] 3 30 5 
[6,] 3 32 
[7,] 4 40 
[8,] 4 45 
[9,] 1 5 

나는이 꽤 R에 새로운 고민입니다. 미리 감사드립니다.

Answer 1

이 rle 및 tapply위한 작업 같은 소리 :

b = c(1,1,2,2,3,3,4,4,1) 
c = c(10,10,20,20,30,30,40,40,5) 
a <- NULL 
a <- matrix(c(b,c), ncol=2) 

A <- rle(a[, 1])$lengths 
tapply(a[, 2], rep(seq_along(A), A), sum) 
# 1 2 3 4 5 
# 20 40 60 80 5 

설명 :

• rle은 행렬 "a"의 첫 번째 열에있는 항목의 런 길이를 식별합니다.
• rep(seq_along(A), A)을 사용하여 실행 길이에서 tapply에 대한 그룹화 변수를 만듭니다.
• 두 가지를 모두 tapply에 넣으면 원하는 합계를 얻을 수 있습니다.

Answer 2

이게 원하는가요? 나는 깨끗한 base 솔루션이 내기,하지만 난 zoo 패키지에 그것을 rollsum으로 시도해 :

library(zoo) 
mm <- cbind(c(1, 1, 2, 2, 3, 3, 4, 4, 1), c(10, 10, 20, 20, 30, 30, 40, 40, 5)) 

# calculate all lagged sums of column 2 
sums <- rollsum(x = mm[ , 2], k = 2) 

# calculate differences between consecutive numbers in column 1 
diffs <- diff(mm[ , 1]) 

# select sums where diff is 0, i.e. where the two consecutive numbers in column 1 are equal. 
sums2 <- sums[diffs == 0] 

sums2 
# [1] 20 40 60 80