在R中,当从另一列中的分隔文本创建列的名称时,新列名称仅从第1行分配

时间:2018-04-13 19:46:18

标签: r rbind delimited-text columnname do.call

使用R为另一列中的分隔文本创建列的名称,新列的名称仅从第一行获取,其余名称标记为NA。即使他们分配了正确的值。

数据由有效的分号分隔值以及一列中的空值和NULL值组成。我正在尝试创建从每个有效分隔值命名的新列,并将值1分配给找到该列名称的行的新列。

e.g。

  A  B                                       C E Domestic Glue_Sniffing NA NA NA NA NA NA NA
1 1  0 ;Domestic;;Domestic abuse;Glue Sniffing 7        1            NA NA NA NA NA NA NA NA
2 2  4                             ;Drug;Abus; 8       NA            NA  1  1 NA NA NA NA NA
3 3  6          ;Drug;Domestic Abuse;Domestic; 9        1            NA  1 NA NA NA NA NA NA
4 4  5                       ;Alcohol;;Verbal; 5       NA            NA NA NA  1  1 NA NA NA
5 5  7                      ;Shinpads;Abus ; ; 6       NA            NA NA  1 NA NA  1 NA NA
6 6  9                                         7       NA            NA NA NA NA NA NA NA NA
7 7 12                                   ;Ail; 8       NA            NA NA NA NA NA NA NA  1
8 8 10                                         7       NA            NA NA NA NA NA NA NA NA
9 9  9                                       ; 8       NA            NA NA NA NA NA NA NA NA

问题是它只查询第一行的名称。 我使用以下答案作为template

我哪里错了? 代码是根据模板,但修改为添加" = 1"每个分隔元素如下:

#Define a function to take vectors like c("A=1","B=2") and changed them into named vectors like c(A="1", B="2").
createNamedVectors <- function(x) {

    a <- strsplit(x,"=")
    setNames(sapply(a,'[',2), sapply(a,'[',1))

}


tmp.df<-data.frame(
    A = c(1L, 2L, 3L, 4L, 5L, 6L, 7L, 8L, 9), 
    B = c(0L, 4, 6, 5L, 7L, 9L, 12L, 10L, 9), 
    C = c(";Ailment;Drug;Abus;Domestic;Domestic abuse;Glue Sniffing", 
          ";Drug;Abus;", 
          ";Drug;Domestic Abuse;Domestic;",
          ";Alcohol;;Verbal;",
          ";Shinpads; ;",
          "",
          ";Ail;",
          " ",
          ";"),
    D = c(";Vodka=2;Drug;Abus;", 
          ";Drug;Abu;", 
          ";Alcohol;Drug;Verbal;",
          ";Drug;Doms;",
          ";Shinpads;",
          " ",
          "",
          ";Ail;",
          "New"),
    E = c(7L, 8L, 9L, 5L, 6L, 7L, 8L, 7L, 8), 
    stringsAsFactors=T
)

DelimitedNamesOfNewCols <- str_replace_all(as.character(tmp.df$C),"Domestic [Aa]buse","Domestic")
DelimitedNamesOfNewCols <- str_replace_all(DelimitedNamesOfNewCols,";*[[:space:]]*;",";")
DelimitedNamesOfNewCols <- str_replace_all(DelimitedNamesOfNewCols,"^;","")
DelimitedNamesOfNewCols <- str_replace_all(DelimitedNamesOfNewCols,";","=1;")
DelimitedNamesOfNewCols <- str_replace_all(DelimitedNamesOfNewCols,"^[[:space:]]+","DUMMY=;")
DelimitedNamesOfNewCols <- str_replace_all(DelimitedNamesOfNewCols,"[[:space:]]+$","DUMMY=;")
DelimitedNamesOfNewCols <- str_replace_all(DelimitedNamesOfNewCols,"[[:space:]]","_")
DelimitedNamesOfNewCols <- str_replace_all(DelimitedNamesOfNewCols,"^$","DUMMY=;")
DelimitedNamesOfNewCols <- str_replace_all(DelimitedNamesOfNewCols,";$","")



ColsAndValsAsNamedVectors <-lapply(strsplit(DelimitedNamesOfNewCols,";"), createNamedVectors)


#Get list of all column names, then trim and remove NA and blanks
UniqueColumnNames <-unique(unlist(sapply(ColsAndValsAsNamedVectors, names)))
UniqueColumnNames <- stri_trim(UniqueColumnNames)
UniqueColumnNames <- UniqueColumnNames[!is.na(UniqueColumnNames)]
UniqueColumnNames <- UniqueColumnNames[stri_cmp_gt(UniqueColumnNames,"")]


#Extract data from all rows for every column
DataFromRowsForEachColumn <-do.call(rbind, lapply(ColsAndValsAsNamedVectors, '[', UniqueColumnNames))


#Convert everything to numeric
class(DataFromRowsForEachColumn)<-"numeric"

#Rejoin with original data.frame removing column 4
cbind(tmp.df[,-4], DataFromRowsForEachColumn)

通过大量调试语句运行并检查 显示为所有条目正确分配名称和值,直到从以下语句返回

lapply(ColsAndValsAsNamedVectors, '[', UniqueColumnNames)

,可在声明

中找到
DataFromRowsForEachColumn <- do.call(rbind, lapply(ColsAndValsAsNamedVectors, '[', UniqueColumnNames))

但是在DataFromRowsForEachColumn中只有第一行的列名。

我是否需要回到绘图板,我是否犯了一个明显的错误,或者是否有一个我想念的更优雅的解决方案?

感谢所有回复

2 个答案:

答案 0 :(得分:2)

有一种更好的方法可以做到这一点,但是你的代码几乎正常工作,所以让我们把它拿到那里。

正如你所说,一切都很好,直到我们到达这里

# problem line
DataFromRowsForEachColumn <-do.call(rbind, lapply(ColsAndValsAsNamedVectors, '[', UniqueColumnNames))

如果我们只运行lapply,我们可以看到它正常工作,但只有每个列表元素中才会找到找到的列。 rbind不会添加名称,它只会从第一个列表元素中获取名称。

lapply(ColsAndValsAsNamedVectors, '[', UniqueColumnNames)
# [[1]]
#       Ailment          Drug          Abus      Domestic Glue_Sniffing          <NA>          <NA>          <NA>          <NA>          <NA> 
#           "1"           "1"           "1"           "1"            NA            NA            NA            NA            NA            NA 
# 
# [[2]]
# <NA> Drug Abus <NA> <NA> <NA> <NA> <NA> <NA> <NA> 
#   NA  "1"  "1"   NA   NA   NA   NA   NA   NA   NA 
# 
# [[3]]
#     <NA>     Drug     <NA> Domestic     <NA>     <NA>     <NA>     <NA>     <NA>     <NA> 
#       NA      "1"       NA      "1"       NA       NA       NA       NA       NA       NA 
# 
# [[4]]
#    <NA>    <NA>    <NA>    <NA>    <NA> Alcohol  Verbal    <NA>    <NA>    <NA> 
#      NA      NA      NA      NA      NA     "1"     "1"      NA      NA      NA 
# 
# [[5]]
#     <NA>     <NA>     <NA>     <NA>     <NA>     <NA>     <NA> Shinpads     <NA>     <NA> 
#       NA       NA       NA       NA       NA       NA       NA      "1"       NA       NA 

简单的解决方法是只修复结果中的名称:

colnames(DataFromRowsForEachColumn) = UniqueColumnNames

添加该行,我认为您会对最终结果感到满意。

其他评论:

我对你的stri_trim行持怀疑态度 - 在你的管道中似乎为时已晚。我认为你需要提前修剪(如果需要的话)。您似乎用"^[[:space:]]+""[[:space:]]+$"替换线来处理它。你可以用str_trim替换那些2。但是,如果在获得唯一列名称时仍有空格,请在上游进行修复。

答案 1 :(得分:2)

不确定我是否帮助您,但为了从;分隔行创建新列,您可以使用精彩的tidyverse包:

library(tidyverse)

df %>%
  separate_rows(C, sep = ';') %>%
  filter (C != '') %>%
  mutate(new = 1) %>%
  spread(C, new)

这会产生

  A  B                     D E    Abus Ail Ailment Alcohol Domestic Domestic abuse Domestic Abuse Drug Glue Sniffing Shinpads Verbal
1 1  0   ;Vodka=2;Drug;Abus; 7 NA    1  NA       1      NA        1              1             NA    1             1       NA     NA
2 2  4            ;Drug;Abu; 8 NA    1  NA      NA      NA       NA             NA             NA    1            NA       NA     NA
3 3  6 ;Alcohol;Drug;Verbal; 9 NA   NA  NA      NA      NA        1             NA              1    1            NA       NA     NA
4 4  5           ;Drug;Doms; 5 NA   NA  NA      NA       1       NA             NA             NA   NA            NA       NA      1
5 5  7            ;Shinpads; 6  1   NA  NA      NA      NA       NA             NA             NA   NA            NA        1     NA
6 7 12                       8 NA   NA   1      NA      NA       NA             NA             NA   NA            NA       NA     NA
7 8 10                 ;Ail; 7  1   NA  NA      NA      NA       NA             NA             NA   NA            NA       NA     NA