数据框列上的字符串拆分 [英] Strsplit on a column of a data frame

问题描述

我有一个data.frame，其中一个变量是向量(或列表)，如下所示：

MyColumn <- c("A, B,C", "D,E", "F","G")
MyDF <- data.frame(group_id=1:4, val=11:14, cat=MyColumn)

#   group_id val    cat
# 1        1  11 A, B,C
# 2        2  12    D,E
# 3        3  13      F
# 4        4  14      G

我想要一个行数与向量一样多的新数据框

FlatColumn <- unlist(strsplit(MyColumn,split=","))

如下所示：

MyNewDF <- data.frame(group_id=c(rep(1,3),rep(2,2),3,4), val=c(rep(11,3),rep(12,2),13,14), cat=FlatColumn)

#   group_id val cat
# 1        1  11   A
# 2        1  11   B
# 3        1  11   C
# 4        2  12   D
# 5        2  12   E
# 6        3  13   F
# 7        4  14   G

本质上，对于MyColumn(字母A到G)列表中的每个元素，我都希望为该列表分配相应的值。每个因素在MyColumn中只出现一次。

这种重塑/退市/合并有没有巧妙的办法？我在MyDF和的strsplit(MyColumn,split=",")元素的长度行上设计了一个非常麻烦的for-循环。我非常肯定，一定有更优雅的方式。

推荐答案

lst <- strsplit(MyColumn, split = ",")
k <- lengths(lst)    ## expansion size
FlatColumn <- unlist(lst, use.names = FALSE)
MyNewDF <- data.frame(group_id = rep.int(MyDF$group_id, k),
                      val = rep.int(MyDF$val, k),
                      cat = FlatColumn)

#  group_id val cat
#1        1  11   A
#2        1  11   B
#3        1  11   C
#4        2  12   D
#5        2  12   E
#6        3  13   F
#7        4  14   G

这篇关于数据框列上的字符串拆分的文章就介绍到这了，希望我们推荐的答案对大家有所帮助，也希望大家多多支持IT屋！