max/ourMELONS
1
0
Fork 0
Code Issues 1 Pull requests Projects Releases Packages Wiki Activity Actions

Added capacity to handle arrays

Browse source
This commit is contained in:
Waldir Leoncio 2020-03-18 11:24:19 +01:00
parent e823b3a58b
commit 8a73fe4bc1
2 changed files with 21 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

17
R/zeros_ones.R
View file

@ -8,8 +8,15 @@
#' @note Actually works for any `x`, but there's no need to bother imposing
#' validation controls here.
zeros_or_ones <- function(n, x) {
# Expanding n to length 2 if necessary
if (length(n) == 1) n <- c(n, n)
return(matrix(x, n[1], n[2]))
# Returning a matrix or an array
if (length(n) == 2) {
return(matrix(x, n[1], n[2]))
} else {
return(array(x, dim=n))
}
}
#' @title Matrix of zeros
@ -17,8 +24,8 @@ zeros_or_ones <- function(n, x) {
#' the `zeros()` function on Matlab
#' @param n1 number of rows
#' @param n2 number of columns
zeros <- function(n1, n2 = n1) {
return(zeros_or_ones(c(n1, n2), 0))
zeros <- function(n1, n2 = n1, ...) {
return(zeros_or_ones(c(n1, n2, ...), 0))
}
#' @title Matrix of ones
@ -26,6 +33,6 @@ zeros <- function(n1, n2 = n1) {
#' the `ones()` function on Matlab
#' @param n1 number of rows
#' @param n2 number of columns
ones <- function(n1, n2 = n1) {
return(zeros_or_ones(c(n1, n2), 1))
ones <- function(n1, n2 = n1, ...) {
return(zeros_or_ones(c(n1, n2, ...), 1))
}