在数据框中计算每13行的平均值

16

我有一个包含 2 列和 3659 行的数据框 df

我正在尝试通过对这个数据框中的每 10 或 13 行进行平均值来减少数据集大小,因此我尝试了以下代码:

# number of rows per group
n=13
# number of groups
n_grp=nrow(df)/n
round(n_grp,0)
# row indices (one vector per group)
idx_grp <- split(seq(df), rep(seq(n_grp), each = n))

# calculate the col means for all groups
res <- lapply(idx_grp, function(i) {
  # subset of the data frame
  tmp <- dat[i]
  # calculate row means
  colMeans(tmp, na.rm = TRUE)
})
# transform list into a data frame
dat2 <- as.data.frame(res)

然而,我无法将行数除以10或13,因为数据长度不是分割变量的倍数。所以我不确定接下来该怎么做(也许只是想计算最后一组的平均值——即使它只有少于10个元素)。

我也尝试了这个,但结果是相同的:

df1=split(df, sample(rep(1:301, 10)))

今天有人问了类似的问题。结合使用splitcut函数。看看这个链接是否有帮助:https://dev59.com/yYvda4cB1Zd3GeqPUhNo - Pierre L
我遇到的问题是无法按照我想要的组数进行分割。因为我的列数是奇数,例如我想要有10个组。 - yuliaUU
4个回答

17

这里有一个使用 aggregate()rep() 的解决方案。

df <- data.frame(a=1:12, b=13:24 );
df;
##     a  b
## 1   1 13
## 2   2 14
## 3   3 15
## 4   4 16
## 5   5 17
## 6   6 18
## 7   7 19
## 8   8 20
## 9   9 21
## 10 10 22
## 11 11 23
## 12 12 24
n <- 5;
aggregate(df, list(rep(1:(nrow(df) %/% n + 1), each = n, len = nrow(df))), mean)[-1];
##      a    b
## 1  3.0 15.0
## 2  8.0 20.0
## 3 11.5 23.5

这个解决方案的重要部分是处理 nrow(df) 不能被 n 整除的问题,其中需要指定 rep() 函数的 len 参数(实际上完整的参数名是 length.out),它会自动将组向量限制在适当的长度。


[-1]在末尾是什么意思?删除添加的额外列?这对我来说很奇怪,我正在使用一个包含DF列表的for循环。 - Garglesoap

10
如果 df 是一个 data.table,你可以使用 %/% 来进行分组,例如:
library(data.table)
setDT(df)
n <- 13 # every 13 rows

df[, mean(z), by= (seq(nrow(df)) - 1) %/% n]

如果你需要每n行输出一次,那么请使用%%代替%/%
df[, mean(z), by= (seq(nrow(df)) - 1) %% n]

我喜欢在data.table中使用编程方式!只是一个提示,(seq(nrow(df))可以被替换为(1:.N)以更符合data.table的方式,如果整个内容被包含在更大的分组中,例如 df[,by = my.group, .SD[, by = (1:.N - 1) %/% n, mean (z)] ],这也可以起作用。 - JVP

7

这应该可行。使用 n = 13 将 13 行分组在一起。如果你有 27 行,你将得到大小为 13、13、1 的三个组。

n.colmeans = function(df, n = 10){
    aggregate(x = df,
              by = list(gl(ceiling(nrow(df)/n), n)[1:nrow(df)]),
              FUN = mean)
}

n.colmeans(state.x77, 10)

  Group.1 Population Income Illiteracy Life Exp Murder HS Grad Frost     Area
1       1     4892.8 4690.8       1.44   70.508   9.53   53.63  75.1 116163.6
2       2     3570.5 4419.4       1.12   71.110   7.07   53.35  99.8  44406.6
3       3     3697.9 4505.5       0.93   70.855   6.64   55.25 131.7  60873.0
4       4     5663.9 4331.2       1.33   70.752   7.12   49.59 103.6  56949.5
5       5     3407.0 4232.1       1.03   71.168   6.53   53.72 112.1  75286.7

2
方式
n1 <- 10
iris %>% group_by(mean = (row_number() -1) %/% n1) %>%
  mutate(mean = mean(Sepal.Length))

# A tibble: 150 x 6
# Groups:   mean [15]
   Sepal.Length Sepal.Width Petal.Length Petal.Width Species  mean
          <dbl>       <dbl>        <dbl>       <dbl> <fct>   <dbl>
 1          5.1         3.5          1.4         0.2 setosa   4.86
 2          4.9         3            1.4         0.2 setosa   4.86
 3          4.7         3.2          1.3         0.2 setosa   4.86
 4          4.6         3.1          1.5         0.2 setosa   4.86
 5          5           3.6          1.4         0.2 setosa   4.86
 6          5.4         3.9          1.7         0.4 setosa   4.86
 7          4.6         3.4          1.4         0.3 setosa   4.86
 8          5           3.4          1.5         0.2 setosa   4.86
 9          4.4         2.9          1.4         0.2 setosa   4.86
10          4.9         3.1          1.5         0.1 setosa   4.86
# ... with 140 more rows

如果n1不是df的行数的约数,则也是如此。

n1 <- 7

# A tibble: 150 x 6
# Groups:   mean [21]
   Sepal.Length Sepal.Width Petal.Length Petal.Width Species  mean
          <dbl>       <dbl>        <dbl>       <dbl> <fct>   <dbl>
 1          5.1         3.5          1.4         0.2 setosa    4.9
 2          4.9         3            1.4         0.2 setosa    4.9
 3          4.7         3.2          1.3         0.2 setosa    4.9
 4          4.6         3.1          1.5         0.2 setosa    4.9
 5          5           3.6          1.4         0.2 setosa    4.9
 6          5.4         3.9          1.7         0.4 setosa    4.9
 7          4.6         3.4          1.4         0.3 setosa    4.9
 8          5           3.4          1.5         0.2 setosa    4.8
 9          4.4         2.9          1.4         0.2 setosa    4.8
10          4.9         3.1          1.5         0.1 setosa    4.8
# ... with 140 more rows

您可以在多个列上进行变异。
mydf <- iris[-5]

mydf %>% group_by(n = (row_number() -1) %/% n1) %>%
  mutate(across(everything(), ~ mean(.), .names = "{.col}_mean"))

# A tibble: 150 x 9
# Groups:   n [22]
   Sepal.Length Sepal.Width Petal.Length Petal.Width     n Sepal.Length_me~ Sepal.Width_mean Petal.Length_me~
          <dbl>       <dbl>        <dbl>       <dbl> <dbl>            <dbl>            <dbl>            <dbl>
 1          5.1         3.5          1.4         0.2     0              4.9             3.39             1.44
 2          4.9         3            1.4         0.2     0              4.9             3.39             1.44
 3          4.7         3.2          1.3         0.2     0              4.9             3.39             1.44
 4          4.6         3.1          1.5         0.2     0              4.9             3.39             1.44
 5          5           3.6          1.4         0.2     0              4.9             3.39             1.44
 6          5.4         3.9          1.7         0.4     0              4.9             3.39             1.44
 7          4.6         3.4          1.4         0.3     0              4.9             3.39             1.44
 8          5           3.4          1.5         0.2     1              4.8             3.21             1.43
 9          4.4         2.9          1.4         0.2     1              4.8             3.21             1.43
10          4.9         3.1          1.5         0.1     1              4.8             3.21             1.43
# ... with 140 more rows, and 1 more variable: Petal.Width_mean <dbl>

网页内容由stack overflow 提供, 点击上面的
可以查看英文原文,
原文链接