Как выполнить расчеты после разбиения набора данных на несколько наборов данных? - PullRequest
Новая
       62

Как выполнить расчеты после разбиения набора данных на несколько наборов данных?

2 голосов
/ 06 октября 2019

Я хочу взять набор данных и разбить его на несколько наборов данных. Для упрощенной версии проблемы. Реально у меня будет тысячи строк, но я хотел бы упростить задачу для понимания. Предположим, у вас есть следующий код: 

vec = c(1:10)
df = data.frame(vec)
df
   vec
1    1
2    2
3    3
4    4
5    5
6    6
7    7
8    8
9    9
10  10

Я хотел бы разбить этот набор данных на строки по 5 наблюдений каждый, а затем получить среднее значение для каждых 5 строк.

Пока япопытался разделить код следующим образом: 

splitdf = split(df, rep(1:2,each = 5))

Теперь я хотел бы получить среднее значение для каждой группы. Например, среднее значение для первого фрагмента равно 3, а для второго - 8.

Затем я хотел бы сделать функцию повторения и сохранить ее в отдельном столбце. Я хочу, чтобы мой фрейм данных выглядел следующим образом: 

   vec  mean
1    1     3
2    2     3
3    3     3
4    4     3
5    5     3
6    6     8
7    7     8
8    8     8
9    9     8
10  10     8

Мне было интересно, подойдет ли функция цикла или есть более простой способ решения этой проблемы. Я открыт для предложений.

Ответы [ 3 ]

3 голосов
/ 06 октября 2019

Просто добавьте, если вы хотите работать с разделенным фреймом данных, вот как вы можете это сделать. 

# Your vector
vec = c(1:10)

# your dataframe
df = data.frame(vec)

# Your split df 
splitdf = split(df, rep(1:2,each = 5))

# -------------------------------------------------------------------------
#initialize a list (avg) with the size of splitdf 
avg <- vector("list", length(splitdf))
# loop through each list and compute the mean and assign each to avg
for (i in seq_along(splitdf)){
  avg[[i]] <- mean(splitdf[[i]]$vec)
}
# avg
# [[1]]
# [1] 3
# 
# [[2]]
# [1] 8
# unlist avg and create a column mean on df
df$mean <- rep(unlist(avg), each=5)
# df
#     vec mean
# 1    1    3
# 2    2    3
# 3    3    3
# 4    4    3
# 5    5    3
# 6    6    8
# 7    7    8
# 8    8    8
# 9    9    8
# 10  10    8
2 голосов
/ 06 октября 2019

Нет необходимости разбивать данные, если вы используете ту же логику разделения, что и для группы. Например, в ave

df$mean <- ave(df$vec, rep(1:2,each = 5)) 
df

#   vec mean
#1    1    3
#2    2    3
#3    3    3
#4    4    3
#5    5    3
#6    6    8
#7    7    8
#8    8    8
#9    9    8
#10  10    8

Функция по умолчанию в ave уже равна mean, поэтому мы не применяем ее здесь явно.

1 голос
/ 06 октября 2019

Вы, вероятно, ищете by(), который в основном предлагает функцию раздельного применения. Отключить, используя rbind(). 

res <- do.call(rbind, 
               by(DF, rep(1:2, each=5), function(x) 
                 cbind(x, mean=colMeans(x))  # perform calculations on subsets
                 )
               )
res
#      vec mean
# 1.1    1    3
# 1.2    2    3
# 1.3    3    3
# 1.4    4    3
# 1.5    5    3
# 2.6    6    8
# 2.7    7    8
# 2.8    8    8
# 2.9    9    8
# 2.10  10    8

Данные 

DF <- structure(list(vec = 1:10), class = "data.frame", row.names = c(NA, 
-10L))
...