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
|
library("matrixStats")
rowProds_R <- function(x, FUN = prod, na.rm = FALSE, ..., useNames = TRUE) {
res <- apply(x, MARGIN = 1L, FUN = FUN, na.rm = na.rm)
if (!useNames) names(res) <- NULL
res
}
all.equal.na <- function(target, current, ...) {
# Computations involving NaN may return NaN or NA, cf. ?is.nan
current[is.nan(current)] <- NA_real_
target[is.nan(target)] <- NA_real_
all.equal(target, current, ...)
}
for (mode in c("integer", "double")) {
# Missing values
x <- matrix(c(1, NA, NaN, 1, 1, 0, 1, 0), nrow = 4, ncol = 2)
cat("mode: ", mode, "\n", sep = "")
storage.mode(x) <- mode
str(x)
# To check names attribute
dimnames <- list(letters[1:4], LETTERS[1:2])
# Test with and without dimnames on x
for (setDimnames in c(TRUE, FALSE)) {
if (setDimnames) dimnames(x) <- dimnames
else dimnames(x) <- NULL
# Check names attribute
for (useNames in c(TRUE, FALSE)) {
y0 <- rowProds_R(x, na.rm = TRUE, useNames = useNames)
print(y0)
y1 <- rowProds(x, na.rm = TRUE, useNames = useNames)
print(y1)
y2 <- colProds(t(x), na.rm = TRUE, useNames = useNames)
print(y2)
stopifnot(all.equal(y1, y0))
stopifnot(all.equal(y2, y1))
# Missing values
y0 <- rowProds_R(x, na.rm = FALSE, useNames = useNames)
print(y0)
y1 <- rowProds(x, na.rm = FALSE, useNames = useNames)
print(y1)
y2 <- colProds(t(x), na.rm = FALSE, useNames = useNames)
print(y2)
stopifnot(all.equal(y1, y0))
stopifnot(all.equal(y2, y1))
# "Empty" rows
y0 <- rowProds_R(x[integer(0), , drop = FALSE], na.rm = FALSE, useNames = useNames)
print(y0)
y1 <- rowProds(x[integer(0), , drop = FALSE], na.rm = FALSE, useNames = useNames)
print(y1)
y2 <- colProds(t(x[integer(0), , drop = FALSE]), na.rm = FALSE, useNames = useNames)
print(y2)
stopifnot(all.equal.na(y1, y0))
stopifnot(all.equal(y2, y1))
stopifnot(length(y1) == 0L)
# Using product()
y1 <- rowProds(x, method = "expSumLog", na.rm = FALSE, useNames = useNames)
print(y1)
y2 <- colProds(t(x), method = "expSumLog", na.rm = FALSE, useNames = useNames)
print(y2)
stopifnot(all.equal(y2, y1))
}
}
} # for (mode ...)
# Bug report 2012-06-25
x <- matrix(c(1, 1, 1, 1, 1, 0, 1, 0), nrow = 4, ncol = 2)
y0 <- rowProds_R(x)
print(y0)
y1 <- rowProds(x)
print(y1)
y2 <- colProds(t(x))
print(y2)
stopifnot(all.equal.na(y1, y0))
stopifnot(all.equal.na(y1, x[, 1] * x[, 2]))
stopifnot(all.equal.na(y2, y1))
# Check names attribute
dimnames(x) <- dimnames
y0 <- rowProds_R(x, useNames = TRUE)
print(y0)
y1 <- rowProds(x, useNames = TRUE)
print(y1)
y2 <- colProds(t(x), useNames = TRUE)
print(y2)
stopifnot(all.equal.na(y1, y0))
stopifnot(all.equal.na(y1, x[, 1] * x[, 2]))
stopifnot(all.equal.na(y2, y1))
# Bug report 2014-03-25 ("all rows contains a zero")
x <- matrix(c(0, 1, 1, 0), nrow = 2, ncol = 2)
# To check names attribute
dimnames <- list(letters[1:2], LETTERS[1:2])
y0 <- rowProds_R(x)
print(y0)
y1 <- rowProds(x)
print(y1)
y2 <- colProds(t(x))
print(y2)
stopifnot(all.equal.na(y1, y0))
stopifnot(all.equal.na(y1, c(0, 0)))
stopifnot(all.equal.na(y2, y1))
# Check names attribute
dimnames(x) <- dimnames
y0 <- rowProds_R(x, useNames = TRUE)
print(y0)
y1 <- rowProds(x, useNames = TRUE)
print(y1)
y2 <- colProds(t(x), useNames = TRUE)
print(y2)
stopifnot(all.equal.na(y1, y0))
stopifnot(all.equal.na(y2, y1))
|