2017-08-28 48 views
9

比方說,我有一個數據幀有幾行類似如下:選擇值按行根據排名日期間

df <- data.frame(a = c(NA,20,NA), 
       date1 = c("2016-03-01", "2016-02-01", "2016-02-01"), 
       b = c(50,NA, NA), 
       date2 = c("2016-02-01", "2016-03-01", "2016-03-01"), 
       c = c(10,10, 10), 
       date3 = c("2016-01-01","2016-01-01", "2016-01-01")) 

對於每一行,我想這不是一個NA最新值在a,b,c之間根據dates(所以我分別看date1,date2date3並選擇最近的一個)。

基本上,date1給出對應於值adate2日期給出對應於值b日期, date3給出對應於值c的日期。

如果date1 > date2 & date1 > date3,我將要採取的價值a 然而,如果該值aNA(這是在我的例子的情況下),我會比較date2date3。在我的示例中,date2 > date3以及因爲值b不是NA而是50,因此我將採用50作爲我的最終結果。

現在我想既然我使用dplyr在我的數據幀

所有行做到這一點,我嘗試使用排序功能(在我的例子使用case_when功能,我期待一個第一排日期,然後看鏈接的價值。如果它是一個不適用的,我看看排名第二的等等......)

但是,我不能只是說,因爲我想要做的, :

df <- df %>% 
     mutate(result = case_when(is.na(a) & is.na(b) & is.na(c) ~ NA_integer_, 
            rev(rank(date1, date2, date3))[1] == 3 & !is.na(a) ~ a, 
            rev(rank(date1, date2, date3))[2] == 3 & !is.na(b) ~ b, 
            rev(rank(date1, date2, date3))[3] == 3 & !is.na(a) ~ c, 
            rev(rank(date1, date2, date3))[1] == 2 & !is.na(a) ~ a, 
            rev(rank(date1, date2, date3))[2] == 2 & !is.na(b) ~ b, 
            rev(rank(date1, date2, date3))[3] == 2 & !is.na(a) ~ c, 
            rev(rank(date1, date2, date3))[1] == 1 & !is.na(a) ~ a, 
            rev(rank(date1, date2, date3))[2] == 1 & !is.na(b) ~ b, 
            rev(rank(date1, date2, date3))[3] == 1 & !is.na(a) ~ c)) 

因爲rank功能需要一個獨特的向量作爲參數(但我不能把c(date1, date2, date3)既不會因爲它會給我這個向量的整個秩序,而不是每行的秩序)

在我的例子中,我想要的結果將是

res 

a date1   b  date2  c date3  result 
NA 2016-03-01 50  2016-02-01 10 2016-01-01 50 
20 2016-02-01 NA  2016-03-01 10 2016-01-01 20 
NA 2016-02-01 NA  2016-03-01 10 2016-01-01 10 

有沒有人有一個想法,甚至完全不同的方法來解決這個問題?

回答

4

我建議轉換爲長格式和計算相關的值。如果你願意,你可以將結果添加到原始數據框中。這裏是你如何使用data.table來做到這一點:

library(data.table) 
setDT(df)      # convert to data.table object 
df[, row := .I]    # add a row-id 
dflong <- melt(df, id = "row", measure = patterns("^date", "^(a|b|c)"), 
       na.rm = TRUE) # convert to long format 
setorder(dflong, value1)  # reorder by date value 
dflong <- unique(dflong, by = "row", fromLast = TRUE) # get the latest dates 
df[dflong, result := i.value2, on = "row"] # add result to original data 

df 
# a  date1 b  date2 c  date3 row result 
#1: NA 2016-03-01 50 2016-02-01 10 2016-01-01 1  50 
#2: 20 2016-02-01 NA 2016-03-01 10 2016-01-01 2  20 
#3: NA 2016-02-01 NA 2016-03-01 10 2016-01-01 3  10 
+0

謝謝!由於使用data.table對象,算法的速度令我印象深刻 – MBB

2

這裏是一個辦法做到這一點......

df$result <- apply(df, 1, function(x){ 
    dates <- as.Date(x[seq(2, length(x), 2)]) 
    values <- x[seq(1,length(x),2)] 
    return(values[!is.na(values)][which.max(dates[!is.na(values)])]) 
}) 

df 
    a  date1 b  date2 c  date3 result 
1 NA 2016-03-01 50 2016-02-01 10 2016-01-01  50 
2 20 2016-02-01 NA 2016-03-01 10 2016-01-01  20 
3 NA 2016-02-01 NA 2016-03-01 10 2016-01-01  10 
3

這應該處理它。首先,我們將數據放在整齊的表單中(每個Date,Value有1行,以及row_num來標識整齊行所屬的示例)。然後我們過濾掉NAs,group_by row_num,按日期降序排序,並取第一行。

df %>% 
    mutate(row_num = row_number()) %>% 
    unite(a, a, date1) %>% 
    unite(b, b, date2) %>% 
    unite(c, c, date3) %>% 
    gather(key, value, -row_num) %>% 
    select(-key) %>% 
    separate(value, into=c("Value", "Date"), sep = "_") %>% 
    mutate(Date = as.Date(Date)) %>% 
    filter(Value != "NA") %>% 
    group_by(row_num) %>% 
    top_n(1, Date) %>% 
    ungroup() 

Results

1

這裏多了一個辦法:

df$row <- 1:nrow(df) 

gather(df, key, date_val, date1, date2, date3, -row) %>% 
    select(-key) %>% 
    gather(key, val, a,b,c) %>% 
    filter(!is.na(val)) %>% 
    group_by(row) %>% 
    mutate(max_date = max(date_val)) %>% 
    filter(date_val == max_date) %>% summarise(result = max(val)) %>% 
    left_join(df, by="row") %>% select(-row) 

# A tibble: 3 × 7 
    result  a  date1  b  date2  c  date3 
    <dbl> <dbl>  <fctr> <dbl>  <fctr> <dbl>  <fctr> 
1  50 NA 2016-03-01 50 2016-02-01 10 2016-01-01 
2  20 20 2016-02-01 NA 2016-03-01 10 2016-01-01 
3  10 NA 2016-02-01 NA 2016-03-01 10 2016-01-01 
1

另一個base替代:

df$id <- 1:nrow(df) 
d2 <- reshape(df, varying = list(seq(1, by = 2, len = (ncol(df) - 1)/2), 
           seq(2, by = 2, len = (ncol(df) - 1)/2)), 
       direction = "long") 

d2 <- with(d2, d2[order(-id, date1, decreasing = TRUE), ]) 

cbind(df, res = tapply(d2$a[!is.na(d2$a)], d2$id[!is.na(d2$a)], `[`, 1)) 
# a  date1 b  date2 c  date3 id res 
# 1 NA 2016-03-01 50 2016-02-01 10 2016-01-01 1 50 
# 2 20 2016-02-01 NA 2016-03-01 10 2016-01-01 2 20 
# 3 NA 2016-02-01 NA 2016-03-01 10 2016-01-01 3 10 
0

聚會結束了,但我只是遇到了這個職位,並決定留下以下內容。這個想法是,我寧願創建一個數據框架並完成這項工作。

out <- data.frame(group = 1:nrow(df), 
        date = as.Date(unlist(df[, grep(x = names(df), "[1-9]")]), 
           "%Y-%m-%d"), 
        result = unlist(df[nchar(names(df)) == 1])) %>% 
     filter(complete.cases(.)) %>% 
     group_by(group) %>% 
     slice(which.max(date)) %>% 
     ungroup 

cbind(df, result = out$result) 

# a  date1 b  date2 c  date3 result 
#1 NA 2016-03-01 50 2016-02-01 10 2016-01-01  50 
#2 20 2016-02-01 NA 2016-03-01 10 2016-01-01  20 
#3 NA 2016-02-01 NA 2016-03-01 10 2016-01-01  10 

如果我使用data.table,我會做以下,這是基於docendo的答案。

setDT(df)[, row := .I] 

out <- melt(df, id = "row", measure = patterns("^date", "^(a|b|c)"), 
      value.name = c("date", "result"), na.rm = TRUE) [, 
       date := as.Date(date, "%Y-%m-%d")][, 
        .SD[which.max(date)], by = row][, c("row", "result")] 

df[out, on = "row"] 

# a  date1 b  date2 c  date3 row result 
#1: 20 2016-02-01 NA 2016-03-01 10 2016-01-01 2  20 
#2: NA 2016-03-01 50 2016-02-01 10 2016-01-01 1  50 
#3: NA 2016-02-01 NA 2016-03-01 10 2016-01-01 3  10