Как поднастроить плоскую таблицу сопряженности в R без потери имен строк и столбцов? - PullRequest
Новая
       14

Как поднастроить плоскую таблицу сопряженности в R без потери имен строк и столбцов?

9 голосов
/ 27 марта 2012

Я использую ftable для создания плоской таблицы сопряженности. Однако, когда я задаю таблицу непредвиденных обстоятельств, R удаляет имена строк и столбцов. Есть ли способ поднастроить таблицу так, чтобы имена строк и столбцов оставались в подмножественной таблице? Вот пример: 

# Create fake data
Group1 = sample(LETTERS[1:3], 20, replace=TRUE)
Group2 = sample(letters[1:3], 20, replace=TRUE)
Year = sample(c("2010","2011","2012"), 20, replace=TRUE)
df1 = data.frame(Group1, Group2, Year)

# Create flat contingency table with column margin
table1 = ftable(addmargins(table(df1$Group1, df1$Group2, df1$Year), margin=3))

# Select rows with sum greater than 2
table2 = table1[table1[ ,4] > 2, ]

> table1
     2010 2011 2012 Sum

A a     0    1    2   3
  b     2    1    0   3
  c     0    0    0   0
B a     0    1    1   2
  b     2    0    0   2
  c     1    0    1   2
C a     0    1    0   1
  b     1    0    2   3
  c     3    0    1   4

> table2
     [,1] [,2] [,3] [,4]
[1,]    0    1    2    3
[2,]    2    1    0    3
[3,]    1    0    2    3
[4,]    3    0    1    4

Обратите внимание, как R преобразовал подмножественную таблицу в матрицу, удалив имена столбцов и оба уровня имен строк. Как я могу сохранить структуру ftable в подмножественной таблице?

Ответы [ 3 ]

4 голосов
/ 28 марта 2012

Рассмотрим работу с data.frame частот.Это гораздо лучшая структура данных для работы, особенно если вы собираетесь ее фильтровать.Вот способ построить его, используя пакет reshape. 

# cast the data into a data.frame
library(reshape)
df1$Freq <- 1
df2 <- cast(df1, Group1 + Group2 ~ Year, fun = sum, value = "Freq")
df2
#   Group1 Group2 2010 2011 2012
# 1      A      a    0    0    1
# 2      A      b    1    1    3
# 3      A      c    0    0    1
# 4      B      a    1    2    0
# 5      B      b    1    1    0
# 6      B      c    0    0    1
# 7      C      a    2    0    1
# 8      C      b    2    0    0
# 9      C      c    0    0    2

# add a column for the `Sum` of frequencies over the years
df2 <- within(df2, Sum <- `2010` + `2011` + `2012`)
df2
#   Group1 Group2 2010 2011 2012 Sum
# 1      A      a    0    0    1   1
# 2      A      b    1    1    3   5
# 3      A      c    0    0    1   1
# 4      B      a    1    2    0   3
# 5      B      b    1    1    0   2
# 6      B      c    0    0    1   1
# 7      C      a    2    0    1   3
# 8      C      b    2    0    0   2
# 9      C      c    0    0    2   2

df2[df2$Sum > 2, ]
#   Group1 Group2 2010 2011 2012 Sum
# 2      A      b    1    1    3   5
# 4      B      a    1    2    0   3
# 7      C      a    2    0    1   3
4 голосов
/ 28 марта 2012

Результатом больше не будет ftable объект, потому что некоторые комбинации отсутствуют.

Но вместо этого у вас может быть матрица с именами строк и столбцов. 

ftable_names <- function(x, which="row.vars") {
  # Only tested in dimensions 1 and 2
  rows <- as.vector(Reduce( 
    function(u,v) t(outer(as.vector(u),as.vector(v),paste)), 
    attr(x, which), 
    "" 
  ))
}
i <- table1[ ,4] > 2
table2 <- table1[i,]
rownames(table2) <- ftable_names(table1, "row.vars")[i]
colnames(table2) <- ftable_names(table1, "col.vars")
table2

#      2010  2011  2012  Sum
# A a     1     2     0    3
# A c     0     0     3    3
# B c     0     3     0    3
# C a     3     1     1    5
3 голосов
/ 28 января 2016

ftable создает «плоские» таблицы сопряженности [по] ... переупорядочивая данные в виде [2D] матрицы.Итак, просто используйте as.matrix для преобразования данных в матрицу перед поднабором (если вы используете as.table напрямую, данные возвращаются в структуру с более высокой размерностью). 

# Create flat contingency table with column margin and variable names
table1 <- ftable(addmargins(table(Group1 = df1$Group1, 
                                 Group2 = df1$Group2, 
                                 Year = df1$Year), margin=3))

# Convert to matrix
mat1 <- as.matrix(table1)
mat2 <- mat1[mat1[ ,4] > 2, ]
mat2

> mat2
             Year
Group1_Group2 2010 2011 2012 Sum
          A_b    3    0    0   3
          A_c    0    2    3   5
          B_b    2    0    1   3

Если вы действительно этого не делаетекак "_", затем замените, используя gsub.

dimnames(mat2) <- rapply(dimnames(mat2), gsub, pattern = "_", replacement = " ", how = "replace")

Редактировать

Или, альтернативно, использовать пакеты dplyr и tidyr для гибкости и удобочитаемости кода: 

library(dplyr)
library(tidyr)

df1 %>% 
  group_by(Group1, Group2, Year) %>%
  tally() %>%
  spread(Year, n, fill = 0) %>%
  ungroup() %>% 
  mutate(Sum = rowSums(.[-(1:2)])) %>%
  filter(Sum > 2) %>%
  unite(Name, c(Group1, Group2), sep = " ")

Source: local data frame [5 x 5]

   Name  2010  2011  2012   Sum
  (chr) (dbl) (dbl) (dbl) (dbl)
1   A a     2     1     0     3
2   A b     1     1     1     3
3   B b     2     0     2     4
4   B c     1     2     0     3
5   C a     1     2     0     3
...