Рассчитать среднее по элементам в списке - PullRequest
Новая
       8

Рассчитать среднее по элементам в списке

2 голосов
/ 18 марта 2019

У меня есть list как это: 

(mylist <- list(a = data.frame(x = c(1, 2), y = c(3, 4)),
                b = data.frame(x = c(2, 3), y = c(4, NA)),
                c = data.frame(x = c(3, 4), y = c(NA, NA))))
$a
  x y
1 1 3
2 2 4

$b
  x  y
1 2  4
2 3 NA

$c
  x  y
1 3 NA
2 4 NA

, который создается purrr::map().Как я могу рассчитать средние значения в соответствующих ячейках?т.е. 

  x   y
1 2 3.5
2 3   4

где 

mean(c(1,  2,  3), na.rm = T) # = 2
mean(c(2,  3,  4), na.rm = T) # = 3
mean(c(3,  4, NA), na.rm = T) # = 3.5
mean(c(4, NA, NA), na.rm = T) # = 4

Спасибо за помощь!

Ответы [ 3 ]

2 голосов
/ 18 марта 2019

Один из методов - преобразовать ваш список в массив, а затем применить функцию среднего к третьему измерению массива: 

my_array <- array(unlist(mylist), dim=c(2,2,3))
apply(my_array, c(1,2), mean, na.rm=T)

#      [,1] [,2]
# [1,]    2  3.5
# [2,]    3  4.0

Если вы хотите сделать все это за один раз, без жесткого кодирования размеров, вы можете сделать: 

apply(array(unlist(mylist), dim=c(nrow(mylist[[1]]),ncol(mylist[[1]]),length(mylist))), c(1,2), mean, na.rm=T)
1 голос
/ 18 марта 2019 
Reduce(function(x, y) x + replace(y, is.na(y), 0), mylist)/
    Reduce(`+`, lapply(mylist, function(x) !is.na(x)))
#  x   y
#1 2 3.5
#2 3 4.0

OR 

nm = c("x", "y")  # could do `nm = names(mylist[[1]])`
sapply(nm, function(NM)
    rowMeans(do.call(cbind, lapply(mylist, function(x) x[NM])), na.rm = TRUE))
#     x   y
#[1,] 2 3.5
#[2,] 3 4.0
1 голос
/ 18 марта 2019

A purrr опция 

library(purrr)
map_df(transpose(mylist), ~rowMeans(as.data.frame(.x), na.rm = TRUE))
 # A tibble: 2 x 2
#      x     y
#  <dbl> <dbl>
#1     2   3.5
#2     3   4
...