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
|
#' Restore the type of columns according to a reference data frame
#'
#' @param data A data frame for which to restore the column types.
#' @inheritParams data_to_long
#' @inheritParams data_rename
#' @param reference A reference data frame from which to find the correct
#' column types. If `NULL`, each column is converted to numeric if it doesn't
#' generate `NA`s. For example, `c("1", "2")` can be converted to numeric but not
#' `c("Sepal.Length")`.
#' @return
#'
#' A data frame with columns whose types have been restored based on the
#' reference data frame.
#'
#' @examples
#' data <- data.frame(
#' Sepal.Length = c("1", "3", "2"),
#' Species = c("setosa", "versicolor", "setosa"),
#' New = c("1", "3", "4")
#' )
#'
#' fixed <- data_restoretype(data, reference = iris)
#' summary(fixed)
#' @export
data_restoretype <- function(data, reference = NULL, ...) {
for (col in names(data)) {
# No reference data (regular fixing) ----------------
if (is.null(reference)) {
if (is.character(data[[col]])) {
data[[col]] <- coerce_to_numeric(data[[col]])
}
} else {
if (is.factor(reference[[col]]) && !is.factor(data[[col]])) {
# Restore factor levels
data[[col]] <- factor(data[[col]], levels = levels(reference[[col]]))
}
if (is.numeric(reference[[col]]) && !is.numeric(data[[col]])) {
data[[col]] <- coerce_to_numeric(as.character(data[[col]]))
}
if (is.character(reference[[col]]) && !is.character(data[[col]])) {
data[[col]] <- as.character(data[[col]])
}
}
}
data
}
|
