Chuyển đổi một biến liên tục được nhóm thành các hàng trong R

Question

Tôi có một khung dữ liệu với các giá trị giả này và tôi muốn thực hiện hồi quy lm trên chúng. Một trong các biến là biến số liên tục được nhóm như được hiển thị bên dưới

df <- data.frame("y" = c(10, 11, 12, 13, 14), 
       "x" = as.factor(c("100-102", "103-105", "106-108", "109-111", "112-114"))) 

Tôi muốn khôi phục y ~ x, Một cách là thay thế các yếu tố x bằng giá trị số trung bình của chúng. Điều này có thể dễ dàng thực hiện bằng cách sử dụng cụm từ thông dụng.

Một cách khác là tạo ra các hàng bổ sung và mở rộng bộ dữ liệu của bạn để nó trông như thế này

data.frame("y" = c(10, 10, 10, 11, 11, 11......), 
      "x" = c(100, 101, 102, 103, 104, 105......)) 

Có một chức năng mà sẽ làm điều này?

Tôi đang nghĩ đến việc tạo các biến bổ sung như x1, x2, x3 và sau đó sử dụng gói reshape2 để chuyển đổi x cột thành hàng.

Answer 1

Giải pháp data.table. Điều này cũng nên thực sự nhanh trên số lớn của data.frame.

require(data.table) 
dt <- data.table(df, key="y") 
dt[, list(x=seq(sub("-.*$", "", x), sub(".*-", "", x))),by=y] 

Nếu bạn có nhiều cột và bạn không muốn mỗi kết hợp khi tách theo cột x, thì đây là mã để sử dụng:

require(data.table) 
dt <- data.table(df) 
# get all column names except "x" 
key.cols <- setdiff(names(df), "x") 
# set the data.table columns to key.cols 
setkeyv(dt, key.cols) 
dt.out <- dt[, list(x=seq(sub("-.*$", "", x), sub(".*-", "", x))), by = key.cols] 

Điều này sẽ cho bạn biết những gì bạn mong đợi.

Answer 2

require(stringr) 
require(foreach) 

foreach(i=1:nrow(df), .combine=rbind) %do% { 
    s <- as.numeric(str_extract_all(df$x[i], "[0-9]+")[[1]]) 
    data.frame(y=rep(df$y[i], s[2]-s[1]+1), x=seq(s[1], s[2])) 
} 

Nếu bạn data.frame thực sự lớn, bạn có thể đi cùng với %dopar%.