如何为H2OFrame对象一次强制多个列到因子?

时间:2018-03-20 20:13:13

标签: r h2o

我正在尝试按照问题提出建议:"Coerce multiple columns to factors at once",但它不适用于H2OFrame对象,例如:

data <- data.frame(matrix(sample(1:40), 4, 10, dimnames = list(1:4, LETTERS[1:10])))
data.hex <- as.h2o(data, destination_frame = "data.hex")
cols <- c("A", "C", "D", "H")
data.hex[cols] <- lapply(data.hex[cols], factor)

产生以下错误消息:

Error in `[<-.H2OFrame`(`*tmp*`, cols, value = list(1L, 1L, 1L, 1L, 1L,  : 
  `value` can only be an H2OFrame object or a numeric or character vector
In addition: 
Warning message:
In if (is.na(value)) value <- NA_integer_ else if (!is.numeric(value) &&  :


the condition has length > 1 and only the first element will be used

如果我试图逐个强制作为因素,它就有效。另一种解决方法是先将data.frame强制为因子,然后将其转换为H2OFrame对象,例如:

data[cols] <- lapply(data[cols], factor)
data.hex <- as.h2o(data, destination_frame = "data.hex")

任何解释为什么会发生或任何更好的解决方法?

1 个答案:

答案 0 :(得分:2)

正确的方法是使用H2OFrame apply()函数,但这会产生与@MKR提到的相同的错误。我创建了一张JIRA票here

理论上,这应该有效:

data.hex[,cols] <- apply(X = data.hex[,cols], MARGIN = 2, FUN = as.factor)

目前,解决方法是:

for (col in cols) {
  data.hex[col] <- as.factor(data.hex[col])
}