如何遍历数据帧列表以在R中设置列名?

问题描述

我有大量的数据帧列表,我想根据第一列和第四行的条件设置列名。请在下面找到我的示例数据:

sample_data<-data.frame(a= c("\"\"","This","Week","1","2","3","4","5","6" ),b=c("\"\"","Last","(1)","(2)","(3)","(4)","(5)","(6)"   ),c=c("\"\"","End","2008","<1>","<2>","<3>","<4>","<5>","<6>"  ),d=c("","Name","","Tiger Woods","Sergio Garcia","Phil Mickelson","padraig Harrington","Vijay Singh","Robert Karlsson"),e=c("","Country","United States","Spain","Ireland","Fiji","Sweden"     ),f=c("","Average","Points","11.664","7.992","6.871","6.832","6.631","5.017"  ),"Total","466.573","415.591","336.674","348.431","358.071","265.906" ))



another_data<-data.frame(a=c("\"\"","\"\"","(50)","(55)","(51)","(52)"),"2011","<148>","<70>","<49>","<51>"),c=c("","Kyle Stanley","Kevin Na","Gonzalo Fdez-Castano","Ryo Ishikawa"),"State","Japan"),e=c("\"\"",f=c("\"\"","2.694","2.560","2.544","2.539")) 
     
               
list_df<-list(sample_data,another_data)


list_df
[[1]]
     a    b    c                  d             e       f     f.1
1   ""   ""   ""                                                 
2 This Last  End               Name       Country Average   Total
3 Week Week 2008                                   Points  Points
4    1  (1)  <1>        Tiger Woods United States  11.664 466.573
5    2  (2)  <2>      Sergio Garcia         Spain   7.992 415.591
6    3  (3)  <3>     Phil Mickelson United States   6.871 336.674
7    4  (4)  <4> padraig Harrington       Ireland   6.832 348.431
8    5  (5)  <5>        Vijay Singh          Fiji   6.631 358.071
9    6  (6)  <6>    Robert Karlsson        Sweden   5.017 265.906

[[2]]
     a     b                    c             d             e       f
1   ""    ""                                               ""      ""
2 Last   End                 Name         State       Country Average
3 Week  2011                   ""            ""            ""  Points
4   ""    ""                                 ""            ""        
5 (50) <148>         Kyle Stanley United States United States   2.694
6 (55)  <70>             Kevin Na United States United States   2.560
7 (51)  <49> Gonzalo Fdez-Castano         Spain         Spain   2.544
8 (52)  <51>         Ryo Ishikawa         Japan         Japan   2.539

所以我要检查的是第四行是否包含""。如果是这样,请合并数据帧第四行到第一行之间的所有内容。另一方面,如果第四行不包含"",则将第三行与数据帧的第一行之间的所有内容合并。

这是我的审判:

for (index in 1:length(list_df))
{

  if(list_df[[index]][4,1]=="")
    {
      list_df[[index]]<-setNames(as.data.frame(list_df[[index]][-(1:4),]),sapply(as.data.frame(list_df[[index]][1:4,paste,collapse = ""))
    }
  else if(list_df[[index]][4,1]!="")
  {
    list_df[[index]]<-setNames(as.data.frame(list_df[[index]][-(1:3),sapply(as.data.frame(list_df[[index]][1:3,collapse = ""))
  }
  {
    return(list_df)
  }

}

但是,这没有给我想要的输出。仅针对第一个条件工作,而不考虑第二个条件。

这是我得到的输出

list_df

   [[1]]
      ThisWeek LastWeek End2008               Name       Country AveragePoints TotalPoints
    4        1      (1)     <1>        Tiger Woods United States        11.664     466.573
    5        2      (2)     <2>      Sergio Garcia         Spain         7.992     415.591
    6        3      (3)     <3>     Phil Mickelson United States         6.871     336.674
    7        4      (4)     <4> padraig Harrington       Ireland         6.832     348.431
    8        5      (5)     <5>        Vijay Singh          Fiji         6.631     358.071
    9        6      (6)     <6>    Robert Karlsson        Sweden         5.017     265.906
    
    [[2]]
         a     b                    c             d             e       f
    1                                                                    
    2 Last   End                 Name       Country       Country Average
    3 Week  2011                                                   Points
    4                                                                    
    5 (50) <148>         Kyle Stanley United States United States   2.694
    6 (55)  <70>             Kevin Na United States United States   2.560
    7 (51)  <49> Gonzalo Fdez-Castano         Spain         Spain   2.544
    8 (52)  <51>         Ryo Ishikawa         Japan         Japan   2.539

但是,所需的输出如下:

list_df

[[1]]
          ThisWeek LastWeek End2008               Name       Country AveragePoints TotalPoints
        4        1      (1)     <1>        Tiger Woods United States        11.664     466.573
        5        2      (2)     <2>      Sergio Garcia         Spain         7.992     415.591
        6        3      (3)     <3>     Phil Mickelson United States         6.871     336.674
        7        4      (4)     <4> padraig Harrington       Ireland         6.832     348.431
        8        5      (5)     <5>        Vijay Singh          Fiji         6.631     358.071
        9        6      (6)     <6>    Robert Karlsson        Sweden         5.017     265.906

[[2]]
   LastWeek   End2011         Name                  State        Country  AveragePoints
5    (50)     <148>         Kyle Stanley           United States United States   2.694
6    (55)      <70>         Kevin Na               United States United States   2.560
7    (51)      <49>         Gonzalo Fdez-Castano   Spain         Spain           2.544
8    (52)      <51>         Ryo Ishikawa           Japan         Japan           2.539

对此我感到非常困惑。我不知道为什么其中一个条件被忽略了。有人可以帮我吗?

解决方法

通常,使用lapply可以很容易地遍历列表。您可以尝试:

lapply(list_df,function(x) {
  if (x[4,1] == '""') {
    names(x) = gsub('"','',trimws(sapply(x[1:4,],paste,collapse = "")))
    x[-c(1:4),]
  }
  else {
    names(x) = gsub('"',trimws(sapply(x[1:3,collapse = "")))
    x[-c(1:3),]
  }
})

#[[1]]
#  ThisWeek LastWeek End2008               Name       Country AveragePoints TotalPoints
#4        1      (1)     <1>        Tiger Woods United States        11.664     466.573
#5        2      (2)     <2>      Sergio Garcia         Spain         7.992     415.591
#6        3      (3)     <3>     Phil Mickelson United States         6.871     336.674
#7        4      (4)     <4> Padraig Harrington       Ireland         6.832     348.431
#8        5      (5)     <5>        Vijay Singh          Fiji         6.631     358.071
#9        6      (6)     <6>    Robert Karlsson        Sweden         5.017     265.906

#[[2]]
#  LastWeek End2011                 Name         State       Country AveragePoints
#5     (50)   <148>         Kyle Stanley United States United States         2.694
#6     (55)    <70>             Kevin Na United States United States         2.560
#7     (51)    <49> Gonzalo Fdez-Castano         Spain         Spain         2.544
#8     (52)    <51>         Ryo Ishikawa         Japan         Japan         2.539

就您的for循环而言,其无效的主要原因是因为您正在寻找错误的字符。 [4,1]处的值为"",不是空字符串(请检查another_data[4,1] == ''another_data[4,1] == '""')。如果您进行更改,则应获得所需的输出。但是,我在您的for循环中做了一些其他更改。

  • 删除了附加的 else if条件,因为它是不需要的。
  • 在我的答案中添加了gsub,以删除列名中的引号
  • return(list_df)没有意义,因为我们已经在list_df循环中更改了for,因此将其删除。
for (index in 1:length(list_df)) {
  if(list_df[[index]][4,1] == '""') {
    list_df[[index]]<-setNames(as.data.frame(list_df[[index]][-(1:4),]),gsub('"',sapply(as.data.frame(list_df[[index]][1:4,collapse = "")))
  }
  else{
    list_df[[index]]<-setNames(as.data.frame(list_df[[index]][-(1:3),sapply(as.data.frame(list_df[[index]][1:3,collapse = "")))
  }
}