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
|
#' @title Initialize data.frame in a convenient way
#'
#' @description
#' Creates a data frame with specified dimensions, column types and optional initial values.
#'
#' @param nrow [\code{integer(1)}]\cr
#' Nubmer of rows.
#' @param ncol [\code{integer(1)}]\cr
#' Number of columns.
#' @param col.types [\code{character(ncol)} | \code{character(1)}]\cr
#' Data types of columns.
#' If you only pass one type, it will be replicated.
#' Supported are all atomic modes also supported by
#' \code{\link[base]{vector}}, i.e. all common data frame types except factors.
#' @param init [any]\cr
#' Scalar object to initialize all elements of the data.frame.
#' You do not need to specify \code{col.types} if you pass this.
#' @param row.names [\code{character} | \code{integer} | \code{NULL}]\cr
#' Row names.
#' Default is \code{NULL}.
#' @param col.names [\code{character} | \code{integer}]\cr
#' Column names.
#' Default is \dQuote{V1}, \dQuote{V2}, and so on.
#' @export
#' @examples
#' print(makeDataFrame(3, 2, init = 7))
#' print(makeDataFrame(3, 2, "logical"))
#' print(makeDataFrame(3, 2, c("logical", "numeric")))
makeDataFrame = function(nrow, ncol, col.types, init, row.names = NULL, col.names = sprintf("V%i", seq_len(ncol))) {
nrow = asCount(nrow)
ncol = asCount(ncol)
if (!missing(col.types))
assertCharacter(col.types, min.len = 1L, any.missing = FALSE)
if (!missing(init)) {
if(!isScalarValue(init))
stop("'init' must be a scalar value!")
if (!missing(col.types)) {
if (length(col.types) > 1L)
stop("If 'init' is given, length of col.types must be 1!")
if (identical(class(init)[1L], "col.types"))
stop("Class of 'init' must match given column type!")
}
}
if (!missing(col.types) && length(col.types) == 1L)
col.types = rep.int(col.types, ncol)
if (!is.null(row.names))
assert(checkIntegerish(row.names, len = nrow), checkCharacter(row.names, len = nrow))
assert(checkIntegerish(col.names, len = ncol), checkCharacter(col.names, len = ncol))
if (nrow == 0L && ncol == 0L)
df = data.frame()
else if (ncol == 0L)
df = data.frame(matrix(nrow = nrow, ncol = 0))
else if (missing(init))
df = lapply(col.types, vector, length = nrow)
else
df = replicate(ncol, rep.int(init, nrow), simplify = FALSE)
df = as.data.frame(df, stringsAsFactors = FALSE)
rownames(df) = row.names
colnames(df) = col.names
return(df)
}
|
