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
|
#' Tools for working with column names
#'
#' @param x A data frame.
#' @param row Row to use as column names.
#' @param na_prefix Prefix to give to the column name if the row has an `NA`.
#' Default is 'x', and it will be incremented at each `NA` (`x1`, `x2`, etc.).
#' @param verbose Toggle warnings.
#' @param prefix Prefix to give to the column name. Default is 'x', and it will
#' be incremented at each column (`x1`, `x2`, etc.).
#'
#' @return
#' `row_to_colnames()` and `colnames_to_row()` both return a data frame.
#'
#' @rdname colnames
#'
#' @export
#'
#' @examples
#' # Convert a row to column names --------------------------------
#' test <- data.frame(
#' a = c("iso", 2, 5),
#' b = c("year", 3, 6),
#' c = c("value", 5, 7)
#' )
#' test
#' row_to_colnames(test)
#'
#' # Convert column names to row --------------------------------
#' test <- data.frame(
#' ARG = c("BRA", "FRA"),
#' `1960` = c(1960, 1960),
#' `2000` = c(2000, 2000)
#' )
#' test
#' colnames_to_row(test)
#'
row_to_colnames <- function(x, row = 1, na_prefix = "x", verbose = TRUE) {
if (!is.numeric(row)) {
insight::format_error("Argument `row` must be of type numeric.")
}
if (length(row) != 1) {
insight::format_error("Argument `row` must be of length 1.")
}
if (nrow(x) < row) {
insight::format_error(
paste0(
"You used row = ", row,
" but the dataset only has ", nrow(x), " rows."
)
)
}
new_colnames <- as.character(unlist(x[row, ], use.names = FALSE))
# Create default colnames if there are NAs in the row used
which_na <- which(is.na(new_colnames))
n_na <- length(which_na)
if (n_na > 0) {
for (i in seq_along(which_na)) {
new_colnames[which_na[i]] <- paste0(na_prefix, i)
}
if (verbose) {
insight::format_warning(
paste0(
"Some values of row ", row,
" were NAs. The corresponding column names are prefixed with `",
na_prefix, "`."
)
)
}
}
colnames(x) <- new_colnames
x[-row, ]
}
#' @rdname colnames
#' @export
colnames_to_row <- function(x, prefix = "x") {
if (length(prefix) != 1) {
insight::format_error("Argument `prefix` must be of length 1.")
}
if (!is.character(prefix)) {
insight::format_error("Argument `prefix` must be of type character.")
}
x2 <- rbind(colnames(x), x)
colnames(x2) <- paste0(prefix, seq_len(ncol(x2)))
x2
}
|
