将用户定义的函数应用于数据帧列表

Question

我有一系列与此类似的数据帧：

df <- data.frame(x = c('notes','year',1995:2005), y = c(NA,'value',11:21))  
df2 <- data.frame(x = c('notes','year',1995:2005), y = c(NA,'value',50:60))

为了清理它们，我编写了一个带有一系列清理步骤的用户定义函数：

clean <- function(df){
  colnames(df) <- df[2,]
  df <- df[grep('^[0-9]{4}', df$year),]
  return(df)
}

我现在想将数据框放在列表中：

df_list <- list(df,df2)

并立即清洁它们。 我试过了

lapply(df_list, clean)

和

for(df in df_list){
  clean(df)
}

但是用这两种方法我都会收到错误：

Error in df[2, ] : incorrect number of dimensions

是什么导致此错误，我该如何解决？ 我对这个问题的解决方法是错误的吗？

Answer 1

您很亲密，但是代码中有一个问题。 由于您在数据框的列中有文本，因此将这些列创建为要素而不是字符。 因此，您的列命名不能提供预期的结果。

#need to specify strings to factors as false
df <- data.frame(x = c('notes','year',1995:2005), y = c(NA,'value',11:21), stringsAsFactors = FALSE)  
df2 <- data.frame(x = c('notes','year',1995:2005), y = c(NA,'value',50:60), stringsAsFactors = FALSE)

clean <- function(df){
  colnames(df) <- df[2,]
  #need to specify the column to select the rows
  df <- df[grep('^[0-9]{4}', df$year),]

  #convert the columns to numeric values
    df[, 1:ncol(df)] <- apply(df[, 1:ncol(df)], 2, as.numeric)

  return(df)
}

df_list <- list(df,df2)
lapply(df_list, clean)