簡體   English   中英

如何在某些條件下使用data.table,使用R進行聚合來計算不同列的均值和中位數

[英]How to calculate mean and median of different columns under some conditions using data.table, aggregation with R

我有四個向量(列)

 x y z  t
 1 1 1 10
 1 1 1 15
 1 4 1 14
 2 3 1 15
 2 2 1 17
 2 1 2 19
 2 4 2 18
 2 4 2 NA
 2 2 2 45
 3 3 2 NA
 3 1 3 59
 4 3 3 23
 4 4 3 45
 4 4 4 74
 5 1 4 86

我知道如何計算x,y和z中每個值的向量t的平均值和中位數。 示例是:

   bar <- data.table(expand.grid(x=unique(data[x %in% c(1,2,3,4,5),x]),
                                 y=unique(data[y %in% c(1,2,3,4),y]),
                                 z=unique(data[z %in% c(1,2,3,4),z])))
   foo <- data[z %in% c(1,2,3,4),list(
    mean.t=mean(t,na.rm=T),
    median.t=median(t,na.rm=T))
   ,by=list(x,y,z)]
   merge(bar[,list(x,y,z)],foo,by=c("x","y","z"),all.x=T)

結果是:

     x y z mean.t median.t
  1: 1 1 1   12.5     12.5
  2: 1 1 2     NA       NA
  3: 1 1 3     NA       NA
  4: 1 1 4     NA       NA
  5: 1 2 1     NA       NA
  ........................
  79: 5 4 3    NA       NA
  80: 5 4 4    NA       NA

但是現在我有一個問題:如何對x,y,z和t進行相同的計算,但對於z不是從1到4的數字,而是針對像這樣的組:

  if 0 < z <= 2 group I, 
  if 2 < z <= 3 group II and 
  if 3 < z <= 4 group III.

因此,輸出應采用以下格式:

     x y z    mean.t median.t
  1: 1 1 I   
  2: 1 1 II     
  3: 1 1 III     
  4: 1 2 I     
  5: 1 2 II     
  6: 1 2 III     
  7: 1 3 I     
  8: 1 3 II     
  9: 1 3 III     
 10: 1 4 I  
 ..........

定義一個新列zGroup進行分組。

(此示例中的數據與您的數據有些不同)

#create some data
dt<-data.table(x=rep(c(1,2),each=4),
               y=rep(c(1,2),each=2,times=2),
               z=rep(c(1,2,3,4),times=2),t=1:8)

#add a zGroup column
dt[0<z & z<=2, zGroup:=1]
dt[2<z & z<=3, zGroup:=2]
dt[3<z & z<=4, zGroup:=3]

#group by unique combinations of x, y, zGroup taking mean and median of t
dt[,list(mean.t=mean(t), median.t=as.double(median(t))), by=list(x,y,zGroup)]

請注意,這將在不將中位數強制為兩倍的情況下發生錯誤。 有關詳細信息,請參見此帖子

暫無
暫無

聲明:本站的技術帖子網頁,遵循CC BY-SA 4.0協議,如果您需要轉載,請注明本站網址或者原文地址。任何問題請咨詢:yoyou2525@163.com.

 
粵ICP備18138465號  © 2020-2024 STACKOOM.COM