Добавить вычисляемый столбец в фрейм данных R - PullRequest
0 голосов
/ 04 июня 2018

Я хочу рассчитать некоторую статистику (среднее, минимальное, максимальное, стандартное отклонение и т. Д.) Некоторых столбцов из фрейма данных и сохранить эти значения как другой фрейм данных.

Вот пример:

>foo

    Col1 Col2 Col3 Col4
1    1    6   10   60
2    2    7   20   70
3    3    8   30   80
4    4    9   40   90
5    5   10   50  100

Например, я хочу сохранить среднее и минимальное значение Col1 и Col3 в кадре данных, например:

>bar

           Col1       Col3
Mean          3         30
Min           1         10

Я хочу сделать это через цикл, что-то вроде этого:

# Result dataframe
bar <- data.frame(Col1 = integer(), Col3 = integer())

variables_for_stats <- c("Col1","Col3")

# I want to do something on the lines of this:
for (z in variables_for_stats){

    # Populate column with required values
    col <- c(mean(foo$z,min(foo$z)) # Throws an error - argument is not numeric or logical: returning NA

    # Add col to 'bar'
    bar$z<- col # Does not work
}

Мой фактический фрейм данных foo в настоящее время имеет ~ 40 столбцов, а фактические переменные_for_stats около 20. Оба они могут измениться, отсюда и желание сделать эточерез цикл и списки.Как мне это сделать?

Ответы [ 4 ]

0 голосов
/ 04 июня 2018

Используя базу R, вы можете сделать что-то вроде:

aggregate( values~ind,stack(foo),function(x)
     c(mean=mean(x),sd=sd(x),min=min(x),max=max(x)))#Write all the functions you want
   ind values.mean  values.sd values.min values.max
1 Col1    3.000000   1.581139   1.000000   5.000000
2 Col2    8.000000   1.581139   6.000000  10.000000
3 Col3   30.000000  15.811388  10.000000  50.000000
4 Col4   80.000000  15.811388  60.000000 100.000000

Если вам нужна только сводная статистика, то:

 library(tidyverse)
 summary(foo)%>%
     data.frame()%>%
     select(-Var1)%>%
     separate(Freq,c("Fun","Val"),":")%>%
     spread(Fun, Val)

       Var2 1st Qu. 3rd Qu. Max.    Mean    Median  Min.   
1      Col1     2       4       5       3       3       1  
2      Col2     7       9      10       8       8       6  
3      Col3    20      40      50      30      30      10  
4      Col4    70      90     100      80      80      60  
0 голосов
/ 04 июня 2018

, если вы заинтересованы в решении tidyverse ...

library(tidyverse)

foo <- tribble(~Col1, ~Col2, ~Col3, ~Col4,
               1,    6,   10,   60,
               2,    7,   20,   70,
               3,    8,   30,   80,
               4,    9,   40,   90,
               5,   10,   50,  100)

foo %>%
  gather(Col, Value) %>% 
  group_by(Col) %>% 
  summarise(Mean = mean(Value), Minimum = min(Value))
#> # A tibble: 4 x 3
#>   Col    Mean Minimum
#>   <chr> <dbl>   <dbl>
#> 1 Col1      3       1
#> 2 Col2      8       6
#> 3 Col3     30      10
#> 4 Col4     80      60

Редактировать Если вы хотите получить результирующий фрейм данных точно так, как вы указали в своем вопросе, то:

foo %>%
  gather(Col, Value) %>% 
  group_by(Col) %>% 
  summarise(Mean = mean(Value),
            Minimum = min(Value)) %>% 
  gather(Func, Value, 2:3) %>% 
  spread(Col, Value) %>% 
  select(Func, Col1, Col3)

# A tibble: 2 x 3
#  Func     Col1  Col3
#  <chr>   <dbl> <dbl>
#1 Mean        3    30
#2 Minimum     1    10
0 голосов
/ 04 июня 2018

Вы можете сделать это с помощью tidyverse инструментов.Фактический расчет - это просто summarise, остальное - просто преобразовать вывод в желаемый формат.

library(tidyverse)
foo <- read_table2(
  "Col1 Col2 Col3 Col4
1    6   10   60
2    7   20   70
3    8   30   80
4    9   40   90
5   10   50  10"
)

bar <- foo %>%
  summarise_at(
    .vars = vars(Col1, Col3),
    .funs = funs(mean, min)
  ) %>%
  gather(stat, value) %>%
  separate(stat, into = c("Col", "Func")) %>%
  spread(Col, value)
bar
#> # A tibble: 2 x 3
#>   Func   Col1  Col3
#>   <chr> <dbl> <dbl>
#> 1 mean      3    30
#> 2 min       1    10

Создано в 2018-06-04 пакетом Представить (v0.2.0).

0 голосов
/ 04 июня 2018

Мы можем зациклить интересующие столбцы и получить mean и min

sapply(foo[c('Col1', 'Col3')], function(x) c(Mean = mean(x), Min =min(x)))
#      Col1 Col3
#Mean    3   30
#Min     1   10

ПРИМЕЧАНИЕ. Решение на основе применения также является циклом.Но это дает больше контроля, чем for цикл в понимании вывода

Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...