1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
|
library("matrixStats")
rowDiffs_R <- function(x, lag = 1L, differences = 1L, ..., useNames = TRUE) {
ncol2 <- ncol(x) - lag * differences
if (ncol2 <= 0) {
y <- matrix(x[integer(0L)], nrow = nrow(x), ncol = 0L)
# Preserve names attribute
if (useNames && !is.null(rownames(x))) rownames(y) <- rownames(x)
return(y)
}
suppressWarnings({
y <- apply(x, MARGIN = 1L, FUN = diff, lag = lag, differences = differences)
})
y <- t(y)
# Preserve dimnames attribute
dim(y) <- c(nrow(x), ncol2)
if (useNames && !is.null(dimnames(x))) {
colnames <- colnames(x)
if (!is.null(colnames)) {
len <- length(colnames)
colnames <- colnames[(len - ncol2 + 1):len]
}
dimnames(y) <- list(rownames(x), colnames)
}
else dimnames(y) <- NULL
y
}
set.seed(0x42)
# - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
# With and without some NAs
# - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
for (mode in c("integer", "double")) {
cat("mode: ", mode, "\n", sep = "")
for (add_na in c(FALSE, TRUE)) {
cat("add_na = ", add_na, "\n", sep = "")
x <- matrix(sample(10 * 8) + 0.1, nrow = 10L, ncol = 8L)
if (add_na) {
x[3:7, c(2, 4)] <- NA_real_
}
storage.mode(x) <- mode
str(x)
dimnames <- list(letters[1:10], LETTERS[1:8])
# Test with and without dimnames on x
for (setDimnames in c(TRUE, FALSE)) {
if (setDimnames) dimnames(x) <- dimnames
else dimnames(x) <- NULL
# Check dimnames attribute
for (useNames in c(TRUE, FALSE)) {
for (lag in 1:4) {
for (differences in 1:3) {
cat(sprintf("mode: %s, lag = %d, differences = %d\n",
mode, lag, differences))
# Row/column ranges
r0 <- rowDiffs_R(x, lag = lag, differences = differences, useNames = useNames)
r1 <- rowDiffs(x, lag = lag, differences = differences, useNames = useNames)
r2 <- t(colDiffs(t(x), lag = lag, differences = differences, useNames = useNames))
stopifnot(all.equal(r1, r0))
stopifnot(all.equal(r2, r0))
stopifnot(all.equal(r1, r2))
}
}
} # for (useNames ...)
} # for (setDimnames ...)
} # for (add_na ...)
} # for (mode ...)
# - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
# All NAs
# - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
for (mode in c("integer", "double")) {
cat("mode: ", mode, "\n", sep = "")
x <- matrix(NA_real_, nrow = 10L, ncol = 5L)
storage.mode(x) <- mode
str(x)
dimnames <- list(letters[1:10], LETTERS[1:5])
# Test with and without dimnames on x
for (setDimnames in c(TRUE, FALSE)) {
if (setDimnames) dimnames(x) <- dimnames
else dimnames(x) <- NULL
# Check dimnames attribute
for (useNames in c(TRUE, FALSE)) {
r0 <- rowDiffs_R(x, useNames = useNames)
r1 <- rowDiffs(x, useNames = useNames)
r2 <- t(colDiffs(t(x), useNames = useNames))
stopifnot(all.equal(r1, r0))
stopifnot(all.equal(r2, r0))
stopifnot(all.equal(r1, r2))
} # for (useNames ...)
} # for (setDimnames ...)
} # for (mode ...)
# - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
# A 1x1 matrix
# - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
x <- matrix(0, nrow = 1L, ncol = 1L)
dimnames <- list("a", "A")
# Test with and without dimnames on x
for (setDimnames in c(TRUE, FALSE)) {
if (setDimnames) dimnames(x) <- dimnames
else dimnames(x) <- NULL
# Check dimnames attribute
for (useNames in c(TRUE, FALSE)) {
r0 <- rowDiffs_R(x, useNames = useNames)
r1 <- rowDiffs(x, useNames = useNames)
r2 <- t(colDiffs(t(x), useNames = useNames))
stopifnot(all.equal(r1, r0))
stopifnot(all.equal(r2, r0))
stopifnot(all.equal(r1, r2))
} # for (useNames ...)
} # for (setDimnames ...)
|
