在多列上应用tidyr :: separate

时间:2017-02-26 03:00:40

标签: r dataframe lapply tidyr

我想迭代数据框中的列并将它们拆分为基于分隔符的列。我正在使用tidyr::separate,当我一次执行一列时,它会起作用。

例如:

df<- data.frame(a = c("5312,2020,1212"), b = c("345,982,284"))

df <- separate(data = df, col = "a", 
                         into = paste("a", c("col1", "col2", "col3"), 
                                      sep = "_"), sep = ",")

返回:

  a_col1 a_col2 a_col3           b
1   5312   2020   1212 345,982,284

当我尝试在df R的每一列上执行相同的操作时,返回错误

例如我用它来循环:

for(col in names(df)){
    df <- separate(data = df, col = col, 
into = paste(col, c("col1", "col2", "col3), 
sep = "_"), sep = ",")
    }

我希望得到以下输出:

  a_col1 a_col2 a_col3 b_col1 b_col2 b_col3
1   5312   2020   1212    345    982    284

但是R返回此错误:

Error in if (!after) c(values, x) else if (after >= lengx) c(x, values) else c(x[1L:after],  : 
  argument is of length zero

是否有另一种方法可以将tidyr::separate应用于数据框中的多个列?

3 个答案:

答案 0 :(得分:3)

您可以将自定义的separate_()电话投放到Reduce()

sep <- function(...) {
    dots <- list(...)
    n <- stringr::str_count(dots[[1]][[dots[[2]]]], "\\d+")
    separate_(..., into = sprintf("%s_col%d", dots[[2]], 1:n))
}

df %>% Reduce(f = sep, x = c("a", "b"))
#   a_col_1 a_col_2 a_col_3 b_col_1 b_col_2 b_col_3
# 1    5312    2020    1212     345     982     284

否则,cSplit也会这样做。

splitstackshape::cSplit(df, names(df))
#     a_1  a_2  a_3 b_1 b_2 b_3
# 1: 5312 2020 1212 345 982 284

答案 1 :(得分:1)

这适用于每列可变数量的分隔符,在单一语法中。演示详细的示例。

library(tidyverse)

df<- data.frame(a = c("5312,2020,1212", "21,4534"), 
                b = c("345,982,284", "324,234,3425,654"),
                c = c('34,89,89', '87866675'))

df
#>                a                b        c
#> 1 5312,2020,1212      345,982,284 34,89,89
#> 2        21,4534 324,234,3425,654 87866675

reduce(seq_along(df), 
       .init = df, 
       ~ .x %>% separate(names(df)[.y], 
                         sep = ',', 
                         into = paste0(names(df)[.y], '_col_' , seq(1 + max(str_count(df[[.y]], ',')))),
                         fill = 'right'
                         )
       )
#>   a_col_1 a_col_2 a_col_3 b_col_1 b_col_2 b_col_3 b_col_4  c_col_1 c_col_2
#> 1    5312    2020    1212     345     982     284    <NA>       34      89
#> 2      21    4534    <NA>     324     234    3425     654 87866675    <NA>
#>   c_col_3
#> 1      89
#> 2    <NA>

reprex package (v2.0.0) 于 2021 年 7 月 19 日创建

答案 2 :(得分:0)

我有同样的询问(学习tidyverse),因此这样做了。注:我想要一个没有分解的解决方案,所以不依赖于了解colnames。

library(tidyverse)

创建输入:

dft <- as_tibble(data.frame(a = c("5312,2020,1212"), b = c("345,982,284")))
df <- as.data.frame(dft)

创建一个空白的tibble来收集输出:

dft0 <- read_csv("a\na")  
dft0 <- dft0[,-1]
dft00 <- dft0

指定要分隔的元素的长度(可以在循环中完成,但我们从查看dft知道);注:如果您有更好的名称,请使用:

leng <- 3

For-loop版本:

for(x in 1:dim(df)[2]){
        dataCol <- dft[,x]
        newCols <- paste(colnames(dataCol)[1], paste("col", 1:leng, sep="") , sep="_")

        dft0 <- cbind(dft0,
                    separate(data = dataCol,
                             col = colnames(dataCol)[1],
                             into = newCols,
                             sep = ","))}

凌乱的sapply版本:

sapp <- sapply(colnames(df),function(ff){
                            separate(as_tibble(df[,ff]),
                                     "value",
                                     letters[1:leng],
                                     sep=",")})

dft00 <- as_tibble(do.call(cbind, sapp))

colnames(dft00) <- as.vector(sapply(colnames(sapp),
                             function(sa){
                                      paste(sa,
                                            rownames(sapp),
                                            sep="_")
                                         }))