3

我对R中的data.table包很陌生。data.table的版本是1.8.2。

我的数据表有大约 2100 万行,所以理想情况下我很想使用数据表方法来解决我的问题,因为它是当今处理大数据的方法。以下是示例数据和代码:

samp_data <- data.frame(user1 = c(24, 24, 24, 56, 75, 75),
                        user2 = c(43, 43, 57, 34, 61, 61),  
                        amount1 = c(1, 4, 3, 2, 6, 8), 
                        amount2 = c(4, 7, 9, 3, 5, 6), 
                        PURCH_DATE_1 = as.Date(c("2012-01-01", "2012-04-29", 
                        "2012-03-02", "2012-06-15", "2012-03-17", "2012-09-25")), 
                        PURCH_DATE_2 = as.Date(c("2012-04-01", "2012-01-25", 
                        "2012-05-21","2012-08-18", "2012-04-03", "2012-10-29")))
samp_data$DIFF_DAYS <- abs(as.numeric(samp_data$PURCH_DATE_1-
samp_data$PURCH_DATE_2))
samp_data_new<-data.table(samp_data)

我想取回一个包含 7 个原始列但在 user1 和 user2 列中有重复对的数据表,将保留 DIFF_DAYS 列中具有最小值的行。

假设我让你们中的一些人对我想要的东西感到困惑,下面的代码包含所需的输出:

samp_data_desired<-data.frame(user1=c(24,24,56,75),user2=c(43,57,34,61),
                              amount1=c(1,3,2,6),amount2=c(4,9,3,5),
                              PURCH_DATE_1=as.Date(c("2012-01-01","2012-03-02",
                              "2012-06-15","2012-03-17")),
                              PURCH_DATE_2=as.Date(c("2012-04-01","2012-05-21",
                              "2012-08-18","2012-04-03")),
                              DIFF_DAYS=c(91,80,64,17))

我知道如何使用 DT[, min(col1), by=user1] 执行简单的聚合来找到每个 user1 的平均值、最小值和最大值,但我无法成功使用唯一或重复的函数。我试过了:

samp_data_check <- data.table(samp_data, key=c("user1", "user2", 
                            "amount1", "amount2", "PURCH_DATE_1",
                            "PURCH_DATE_2"))


samp_data_test <- samp_data_check[, unique(DIFF_DAYS), by=c("user1", 
                  "user2", "amount1", "amount2", "PURCH_DATE_1", 
                  "PURCH_DATE_2")]

以及一些变化,但我感到非常困惑,所以任何帮助都将不胜感激。

4

1 回答 1

6

data.table我能想到的第一种方式(除了将键列设置为之外没有太大关系DIFF_DAYS)。假设你data.table是 DT:

setkey(DT, "DIFF_DAYS")
DT[!duplicated(DT[, c("user1", "user2")])]

另一种方法(更多data.table):

setkey(DT, "user1", "user2", "DIFF_DAYS")
key.DT <- unique(DT[, 1:2])
DT[key.DT, mult = "first"]
于 2013-04-16T12:40:11.547 回答