为数据帧的“类”提供向量

时间:2012-02-09 16:32:24

标签: r

您知道如何为数据框提供名称向量以更改数据框的列名或行名。是否有类似的方法来提供更改数据框中每列类的名称向量?当您使用colClasses读取带有read.table的数据框时,可以执行此操作。如果数据框是在R?

中创建的呢?
DF <- as.data.frame(matrix(rnorm(25), 5, 5))
str(DF)  #all numeric modes

names(DF) <- c("A", "A2", "B", "B2", "Z") #I want something like this for classes
some_classes_function_like_names(DF) <- c(rep("character", 3), rep("factor", 2))

#I can do it like this but this seems inefficient 
DF[, 1:3] <- lapply(DF[, 1:3], as.character)
DF[, 4:5] <- lapply(DF[, 4:5], as.factor)

str(DF)

编辑:由于没有任何意义,我在上面改变了lapply。

编辑2:如果有办法写一个用户定义的功能,那就足够了

2 个答案:

答案 0 :(得分:8)

试试这个:

toCls <- function(x, cls) do.call(paste("as", cls, sep = "."), list(x))
replace(DF,, Map(toCls, DF, cls))

第二个例子。另请尝试此示例(允许NA用于任何不更改其类的列)。我们加载动物园包,因为它提供了一个具有默认原点的as.Date版本,我们定义了我们自己的as.POSIXct2同样避免必须另外指定原点。

library(zoo) # supplies alternate as.Date with a default origin
as.NA <- identity
as.POSIXct2 <- function(x) as.POSIXct(x, origin = "1970-01-01")

cls2 <- c("character", "Date", NA, "factor", "POSIXct2")
replace(DF,, Map(toCls, DF, cls2))

请注意,只有在将数字转换为"Date""POSIXct"时,才会考虑原因和转换"2000-01-01"等字符串时才需要指定原点,因此无论如何都需要指定对于这种情况,我们不需要加载动物园,我们不需要我们自己的版本as.POSIXct

编辑:添加了另一个例子。

答案 1 :(得分:5)

似乎class(x) <- "factor"不起作用,as(x, "factor")也不起作用,所以我不知道直接做你想做的事情。

......但更明确的方式是:

# Coerces data.frame columns to the specified classes
colClasses <- function(d, colClasses) {
    colClasses <- rep(colClasses, len=length(d))
    d[] <- lapply(seq_along(d), function(i) switch(colClasses[i], 
        numeric=as.numeric(d[[i]]), 
        character=as.character(d[[i]]), 
        Date=as.Date(d[[i]], origin='1970-01-01'), 
        POSIXct=as.POSIXct(d[[i]], origin='1970-01-01'), 
        factor=as.factor(d[[i]]),
        as(d[[i]], colClasses[i]) ))
    d
}

# Example usage
DF <- as.data.frame(matrix(rnorm(25), 5, 5))
DF2 <- colClasses(DF, c(rep("character", 3), rep("factor", 2)))
str(DF2)

DF3 <- colClasses(DF, 'Date')
str(DF3)

有两件事:您可以根据需要添加更多案例。该函数的第一行允许您使用单个类名称进行调用。 switch的最后一个“默认”案例会调用as函数,您的里程可能会有所不同。