1回の呼び出しで複数の集計関数をグループごとに複数の変数に適用する


91

次のデータフレームがあります

x <- read.table(text = "  id1 id2 val1 val2
1   a   x    1    9
2   a   x    2    4
3   a   y    3    5
4   a   y    4    9
5   b   x    1    7
6   b   y    4    4
7   b   x    3    9
8   b   y    2    8", header = TRUE)

id1とid2でグループ化されたval1とval2の平均を計算し、同時に各id1-id2の組み合わせの行数を数えたい。各計算を個別に実行できます。

# calculate mean
aggregate(. ~ id1 + id2, data = x, FUN = mean)

# count rows
aggregate(. ~ id1 + id2, data = x, FUN = length)

1回の呼び出しで両方の計算を行うために、

do.call("rbind", aggregate(. ~ id1 + id2, data = x, FUN = function(x) data.frame(m = mean(x), n = length(x))))

ただし、警告とともに出力が文字化けします。

#     m   n
# id1 1   2
# id2 1   1
#     1.5 2
#     2   2
#     3.5 2
#     3   2
#     6.5 2
#     8   2
#     7   2
#     6   2
# Warning message:
#   In rbind(id1 = c(1L, 2L, 1L, 2L), id2 = c(1L, 1L, 2L, 2L), val1 = list( :
#   number of columns of result is not a multiple of vector length (arg 1)

plyrパッケージを使用することもできますが、データセットが非常に大きく、データセットのサイズが大きくなると、plyrが非常に遅くなり(ほとんど使用できなくなります)。

aggregateまたは他の関数を使用して、1回の呼び出しで複数の計算を実行するにはどうすればよいですか?


他にaggregateもあるの回答に記載byしてtapply
RomanLuštrik2012

回答:


152

すべてを1つのステップで実行し、適切なラベルを付けることができます。

> aggregate(. ~ id1+id2, data = x, FUN = function(x) c(mn = mean(x), n = length(x) ) )
#   id1 id2 val1.mn val1.n val2.mn val2.n
# 1   a   x     1.5    2.0     6.5    2.0
# 2   b   x     2.0    2.0     8.0    2.0
# 3   a   y     3.5    2.0     7.0    2.0
# 4   b   y     3.0    2.0     6.0    2.0

これにより、2つのid列と2つの行列列を持つデータフレームが作成されます。

str( aggregate(. ~ id1+id2, data = x, FUN = function(x) c(mn = mean(x), n = length(x) ) ) )
'data.frame':   4 obs. of  4 variables:
 $ id1 : Factor w/ 2 levels "a","b": 1 2 1 2
 $ id2 : Factor w/ 2 levels "x","y": 1 1 2 2
 $ val1: num [1:4, 1:2] 1.5 2 3.5 3 2 2 2 2
  ..- attr(*, "dimnames")=List of 2
  .. ..$ : NULL
  .. ..$ : chr  "mn" "n"
 $ val2: num [1:4, 1:2] 6.5 8 7 6 2 2 2 2
  ..- attr(*, "dimnames")=List of 2
  .. ..$ : NULL
  .. ..$ : chr  "mn" "n"

以下の@ lord.garbageで指摘されているように、これを使用して、「単純な」列を持つデータフレームに変換できます。 do.call(data.frame, ...)

str( do.call(data.frame, aggregate(. ~ id1+id2, data = x, FUN = function(x) c(mn = mean(x), n = length(x) ) ) ) 
    )
'data.frame':   4 obs. of  6 variables:
 $ id1    : Factor w/ 2 levels "a","b": 1 2 1 2
 $ id2    : Factor w/ 2 levels "x","y": 1 1 2 2
 $ val1.mn: num  1.5 2 3.5 3
 $ val1.n : num  2 2 2 2
 $ val2.mn: num  6.5 8 7 6
 $ val2.n : num  2 2 2 2

これは、LHSの複数の変数の構文です。

aggregate(cbind(val1, val2) ~ id1 + id2, data = x, FUN = function(x) c(mn = mean(x), n = length(x) ) )

1
どうもありがとう。ちなみに、1つの列だけを合計するにはどうすれば集計できますか。複数の数値列がある場合は、列を合計したくありません。もちろん、集計の完了後に列を破棄することもできますが、その場合はCPUサイクルがすでに消費されています。
ブロッコリー

グループ化する要素と集計する列のみを指定します。データで負の列インデックスを使用するか、必要な列を数式のLHSに配置してください。(編集を参照してください。)
IRTFM 2012

2
Windows 7マシンでRStudio 0.98.1014を使用しているときに、user2659402が彼のアップデートで言及したバグに遭遇しました。表示されているようにデータフレームをコンソールに出力すると正常に表示されますが、データフレームをdに保存してから、d $ val1.mnにアクセスしようとすると、NULLが返されます。また、view(d)を実行すると、dの形式が正しくありません。アップデートでコードを使用して修正しました。
JHowIX 2014年

4
難しいのは、 "vals"が通常の列ではなく、それぞれ2列の行列として返されるためです。d$val1[ , ""mn"]で構造を確認してみてくださいstr
IRTFM 2014年

5
agg <- aggregate(cbind(val1, val2) ~ id1 + id2, data = x, FUN = function(x) c(mn = mean(x), n = length(x)))を使用して、行列を含む列をデータフレームにバインドできますagg_df <- do.call(data.frame, agg)。こちらもご覧ください
lord.garbage 2014年

30

質問でこれを考えると:

plyrパッケージを使用することもできますが、データセットが非常に大きく、データセットのサイズが大きくなると、plyrが非常に遅くなり(ほとんど使用できなくなります)。

次に、data.table1.9.4+)で次のことを試すことができます。

> DT
   id1 id2 val1 val2
1:   a   x    1    9
2:   a   x    2    4
3:   a   y    3    5
4:   a   y    4    9
5:   b   x    1    7
6:   b   y    4    4
7:   b   x    3    9
8:   b   y    2    8

> DT[ , .(mean(val1), mean(val2), .N), by = .(id1, id2)]   # simplest
   id1 id2  V1  V2 N
1:   a   x 1.5 6.5 2
2:   a   y 3.5 7.0 2
3:   b   x 2.0 8.0 2
4:   b   y 3.0 6.0 2

> DT[ , .(val1.m = mean(val1), val2.m = mean(val2), count = .N), by = .(id1, id2)]  # named
   id1 id2 val1.m val2.m count
1:   a   x    1.5    6.5     2
2:   a   y    3.5    7.0     2
3:   b   x    2.0    8.0     2
4:   b   y    3.0    6.0     2

> DT[ , c(lapply(.SD, mean), count = .N), by = .(id1, id2)]   # mean over all columns
   id1 id2 val1 val2 count
1:   a   x  1.5  6.5     2
2:   a   y  3.5  7.0     2
3:   b   x  2.0  8.0     2
4:   b   y  3.0  6.0     2

このベンチマーク(およびケース)aggregatedata.table確認 するためのタイミングの比較(問題と他の3つの回答すべてを使用)。aggagg.x


12

count列を追加し、で集計してsumから、スケールバックしてを取得できますmean

x$count <- 1
agg <- aggregate(. ~ id1 + id2, data = x,FUN = sum)
agg
#   id1 id2 val1 val2 count
# 1   a   x    3   13     2
# 2   b   x    4   16     2
# 3   a   y    7   14     2
# 4   b   y    6   12     2

agg[c("val1", "val2")] <- agg[c("val1", "val2")] / agg$count
agg
#   id1 id2 val1 val2 count
# 1   a   x  1.5  6.5     2
# 2   b   x  2.0  8.0     2
# 3   a   y  3.5  7.0     2
# 4   b   y  3.0  6.0     2

列名を保持し、単一のcount列を作成するという利点があります。


12

dplyrパッケージを使用すると、を使用してこれを実現できますsummarise_all。この集計機能を使用すると、他の機能(この場合は適用することができますmeanとをn()非グループ化列のそれぞれに):

x %>%
  group_by(id1, id2) %>%
  summarise_all(funs(mean, n()))

それは与える:

     id1    id2 val1_mean val2_mean val1_n val2_n
1      a      x       1.5       6.5      2      2
2      a      y       3.5       7.0      2      2
3      b      x       2.0       8.0      2      2
4      b      y       3.0       6.0      2      2

グループ化されていないすべての列に関数を適用したくない場合は、それらを適用する列を指定するか、summarise_at()関数を使用して不要な列をマイナスで除外します。

# inclusion
x %>%
  group_by(id1, id2) %>%
  summarise_at(vars(val1, val2), funs(mean, n()))

# exclusion
x %>%
  group_by(id1, id2) %>%
  summarise_at(vars(-val2), funs(mean, n()))

10

おそらくあなたはマージしたいですか?

x.mean <- aggregate(. ~ id1+id2, p, mean)
x.len  <- aggregate(. ~ id1+id2, p, length)

merge(x.mean, x.len, by = c("id1", "id2"))

  id1 id2 val1.x val2.x val1.y val2.y
1   a   x    1.5    6.5      2      2
2   a   y    3.5    7.0      2      2
3   b   x    2.0    8.0      2      2
4   b   y    3.0    6.0      2      2

4

を使用して、plyr::each()複数の関数を導入することもできます。

aggregate(cbind(val1, val2) ~ id1 + id2, data = x, FUN = plyr::each(avg = mean, n = length))

1

別のdplyrオプションはacross、現在の開発バージョンの一部です

#devtools::install_github("tidyverse/dplyr")
library(dplyr)

x %>% 
  group_by(id1, id2) %>% 
  summarise(across(starts_with("val"), list(mean = mean, n = length)))

結果

# A tibble: 4 x 4
# Groups:   id1 [2]
  id1   id2   mean$val1 $val2 n$val1 $val2
  <fct> <fct>     <dbl> <dbl>  <int> <int>
1 a     x           1.5   6.5      2     2
2 a     y           3.5   7        2     2
3 b     x           2     8        2     2
4 b     y           3     6        2     2

packageVersion("dplyr")
[1]0.8.99.9000
弊社のサイトを使用することにより、あなたは弊社のクッキーポリシーおよびプライバシーポリシーを読み、理解したものとみなされます。
Licensed under cc by-sa 3.0 with attribution required.