R:汇总多列(数字,字符)并删除NA

时间:2016-01-18 14:27:04

标签: r dplyr na.rm

我有一个包含许多列(~50)的data.frame。其中一些是字符,一些是数字,其中三个用于分组。

我需要:

  • 从数字列中删除NA
  • 计算每个数字列的平均值
  • 提取字符列的第一个元素

让我们说,我们使用修改后的虹膜数据如下:

data(iris)
iris$year <- rep(c(2000,3000),each=25) ## for grouping
iris$color <- rep(c("red","green","blue"),each=50) ## character column
iris[1,] <- NA ## introducing NAs

我总共有50列,数字和字符混合在一起。我一直在尝试这样的事情:

giris <- group_by(iris, Species, year)
cls <- unlist(sapply(giris, class)) ## find out classes
action <- ifelse(cls == "numeric", "mean", "first")
action <- paste(action)
summarise_each(giris, action)

我得到的是对于组中所有列的含义,后跟具有相应组中第一个值的列。并且没有处理NAs ......这不是我寻求的......

帮助任何人?

2 个答案:

答案 0 :(得分:11)

您可以使用if else中的funs / summarise_each来尝试此操作:

iris %>% 
  group_by(Species, year) %>% 
  summarise_each(funs(if(is.numeric(.)) mean(., na.rm = TRUE) else first(.)))

由于您在分组列中也有一些NA,因此您可以添加filter语句:

iris %>% 
  filter(!is.na(Species) & !is.na(year)) %>% 
  group_by(Species, year) %>% 
  summarise_each(funs(if(is.numeric(.)) mean(., na.rm = TRUE) else first(.)))
#Source: local data frame [6 x 7]
#Groups: Species [?]
#
#     Species  year Sepal.Length Sepal.Width Petal.Length Petal.Width color
#      (fctr) (dbl)        (dbl)       (dbl)        (dbl)       (dbl) (chr)
#1     setosa  2000        5.025    3.479167       1.4625       0.250   red
#2     setosa  3000        4.984    3.376000       1.4640       0.244   red
#3 versicolor  2000        6.012    2.776000       4.3120       1.344 green
#4 versicolor  3000        5.860    2.764000       4.2080       1.308 green
#5  virginica  2000        6.576    2.928000       5.6400       2.044  blue
#6  virginica  3000        6.600    3.020000       5.4640       2.008  blue

为避免颜色列中的潜在NA(或任何非数字列),您可以将其修改为first(na.omit(.))

您也可以尝试data.table

library(data.table)
setDT(iris)
iris[!is.na(Species) & !is.na(year), lapply(.SD, function(x) {
     if(is.numeric(x)) mean(x, na.rm = TRUE) else x[!is.na(x)][1L]}), 
     by = list(Species, year)]
#      Species year Sepal.Length Sepal.Width Petal.Length Petal.Width color
#1:     setosa 2000        5.025    3.479167       1.4625       0.250   red
#2:     setosa 3000        4.984    3.376000       1.4640       0.244   red
#3: versicolor 2000        6.012    2.776000       4.3120       1.344 green
#4: versicolor 3000        5.860    2.764000       4.2080       1.308 green
#5:  virginica 2000        6.576    2.928000       5.6400       2.044  blue
#6:  virginica 3000        6.600    3.020000       5.4640       2.008  blue

答案 1 :(得分:0)

我试一试:

1。 对于你提到的第一点,我会做类似下面的事情(第二点不需要):

na.omit(iris[ , which(sapply(iris, class) == "numeric")])

要将列{i} numericcharacter分开,我使用以下内容:

iris[ , which(sapply(iris, class) == "numeric")]
iris[ , which(sapply(iris, class) == "character")]

2。 第二项任务是将上面的行与colMeans结合起来:

colMeans(iris[ , which(sapply(iris, class) == "numeric")], na.rm = TRUE)

3。 要提取字符列的第一个元素,您只需执行以下操作:

iris[1, which(sapply(iris, class) == "character")]

在上面提到的虹膜数据的情况下,第一行是完全NA,甚至是字符列,所以我会迭代找到第一个非NA行

k <- 1
while(any(is.na(FirstCharacterElement <- iris[k, which(sapply(iris, class) == "character")]))){
  k <- k + 1
}

小心类因子(在iris数据的情况下分解代码,其中列Species是类因子,你可能希望它是一个字符列。你可以用sapply(iris, class)来检查并用例如

进行更改
iris$Species <- as.character(iris$Species) #or with similar column names

当您读入数据时,您可以提及函数stringsAsFactors = FALSEread.table或类似的参数read.csv