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
|
na.roughfix <- function(object, ...)
UseMethod("na.roughfix")
na.roughfix.data.frame <- function(object, ...) {
isfac <- sapply(object, is.factor)
isnum <- sapply(object, is.numeric)
if (any(!(isfac | isnum)))
stop("na.roughfix only works for numeric or factor")
roughfix <- function(x) {
if (any(is.na(x))) {
if (is.factor(x)) {
freq <- table(x)
x[is.na(x)] <- names(freq)[which.max(freq)]
} else {
x[is.na(x)] <- median(x, na.rm=TRUE)
}
}
x
}
object[] <- lapply(object, roughfix)
object
}
na.roughfix.default <- function(object, ...) {
if (!is.atomic(object))
return(object)
d <- dim(object)
if (length(d) > 2)
stop("can't handle objects with more than two dimensions")
if (all(!is.na(object)))
return(object)
if (!is.numeric(object))
stop("roughfix can only deal with numeric data.")
if (length(d) == 2) {
hasNA <- which(apply(object, 2, function(x) any(is.na(x))))
for (j in hasNA)
object[is.na(object[, j]), j] <- median(object[, j], na.rm=TRUE)
} else {
object[is.na(object)] <- median(object, na.rm=TRUE)
}
object
}
|
