create matrix where all diagonals are the same value in anti-diagonal direction

user_4083743 0
Viewed 88

I'm sure theres a name for this type of matrix but not sure of it. In R I want to quickly convert a variable x:

x = 1:10
x  #[1]  1  2  3  4  5  6  7  8  9 10

into a matrix where all the diagonals (not the main diagonal) are the same in one direction, the anti-diagonal direction:

    x
1   1       2       3       4       5       6       7       8       9      10
2   2       3       4       5       6       7       8       9      10      NA
3   3       4       5       6       7       8       9      10      NA      NA
4   4       5       6       7       8       9      10      NA      NA      NA
5   5       6       7       8       9      10      NA      NA      NA      NA
6   6       7       8       9      10      NA      NA      NA      NA      NA
7   7       8       9      10      NA      NA      NA      NA      NA      NA
8   8       9      10      NA      NA      NA      NA      NA      NA      NA
9   9      10      NA      NA      NA      NA      NA      NA      NA      NA
10 10      NA      NA      NA      NA      NA      NA      NA      NA      NA

thanks

r matrix
2 Answers

We can use shift from data.table

library(data.table)
do.call(cbind, shift(x, 0:9, type = 'lead'))
#      [,1] [,2] [,3] [,4] [,5] [,6] [,7] [,8] [,9] [,10]
# [1,]    1    2    3    4    5    6    7    8    9    10
# [2,]    2    3    4    5    6    7    8    9   10    NA
# [3,]    3    4    5    6    7    8    9   10   NA    NA
# [4,]    4    5    6    7    8    9   10   NA   NA    NA
# [5,]    5    6    7    8    9   10   NA   NA   NA    NA
# [6,]    6    7    8    9   10   NA   NA   NA   NA    NA
# [7,]    7    8    9   10   NA   NA   NA   NA   NA    NA
# [8,]    8    9   10   NA   NA   NA   NA   NA   NA    NA
# [9,]    9   10   NA   NA   NA   NA   NA   NA   NA    NA
#[10,]   10   NA   NA   NA   NA   NA   NA   NA   NA    NA

In base R, we can use embed

out <- embed(c(x, x), 10)
replace(out, lower.tri(out), NA)[, 10:1]
#      [,1] [,2] [,3] [,4] [,5] [,6] [,7] [,8] [,9] [,10]
# [1,]    1    2    3    4    5    6    7    8    9    10
# [2,]    2    3    4    5    6    7    8    9   10    NA
# [3,]    3    4    5    6    7    8    9   10   NA    NA
# [4,]    4    5    6    7    8    9   10   NA   NA    NA
# [5,]    5    6    7    8    9   10   NA   NA   NA    NA
# [6,]    6    7    8    9   10   NA   NA   NA   NA    NA
# [7,]    7    8    9   10   NA   NA   NA   NA   NA    NA
# [8,]    8    9   10   NA   NA   NA   NA   NA   NA    NA
# [9,]    9   10   NA   NA   NA   NA   NA   NA   NA    NA
#[10,]   10   NA   NA   NA   NA   NA   NA   NA   NA    NA
#[11,]   NA   NA   NA   NA   NA   NA   NA   NA   NA    NA
user_37322710

Here the alternative solution is using base R

f <- function(x) {
  y <- x:10
  length(y) <- 10
  return(y)
}

sapply(x, f)

     [,1] [,2] [,3] [,4] [,5] [,6] [,7] [,8] [,9] [,10]
 [1,]    1    2    3    4    5    6    7    8    9    10
 [2,]    2    3    4    5    6    7    8    9   10    NA
 [3,]    3    4    5    6    7    8    9   10   NA    NA
 [4,]    4    5    6    7    8    9   10   NA   NA    NA
 [5,]    5    6    7    8    9   10   NA   NA   NA    NA
 [6,]    6    7    8    9   10   NA   NA   NA   NA    NA
 [7,]    7    8    9   10   NA   NA   NA   NA   NA    NA
 [8,]    8    9   10   NA   NA   NA   NA   NA   NA    NA
 [9,]    9   10   NA   NA   NA   NA   NA   NA   NA    NA
[10,]   10   NA   NA   NA   NA   NA   NA   NA   NA    NA

This is much easier to understand, and does not require any package.

user_50473110
Related