Fixed bugs in size() for non-sequential x

This commit is contained in:
Waldir Leoncio 2020-03-03 14:31:15 +01:00
parent 8ecc59a4e6
commit 11bdefab7c

View file

@ -25,12 +25,12 @@ size <- function(x, d) {
n_dim <- ifelse(is.null(dim(x)), 1, length(dim(x))) n_dim <- ifelse(is.null(dim(x)), 1, length(dim(x)))
if (missing(d)) { if (missing(d)) {
if (n_dim == 1) { if (n_dim == 1) {
out <- range(x) out <- c(1, length(x))
} else { } else {
out <- dim(x) out <- dim(x)
} }
} else { } else {
out <- ifelse(n_dim == 1, range(x)[d], dim(x)[d]) out <- ifelse(n_dim == 1, c(1, length(x))[d], dim(x)[d])
if (is.na(out)) out <- 1 # for MATLAB compatibility if (is.na(out)) out <- 1 # for MATLAB compatibility
} }
return(out) return(out)