Разделить столбец строки фрейма данных на несколько столбцов - PullRequest
201 голосов
/ 04 декабря 2010

Я хотел бы взять данные вида

before = data.frame(attr = c(1,30,4,6), type=c('foo_and_bar','foo_and_bar_2'))
  attr          type
1    1   foo_and_bar
2   30 foo_and_bar_2
3    4   foo_and_bar
4    6 foo_and_bar_2

и использовать split() в столбце "type" сверху, чтобы получить что-то вроде этого:

  attr type_1 type_2
1    1    foo    bar
2   30    foo  bar_2
3    4    foo    bar
4    6    foo  bar_2

Я придумал что-то невероятно сложное, включающее какую-то форму apply, которая работала, но с тех пор я потерял это место.Это казалось слишком сложным, чтобы быть лучшим способом.Я могу использовать strsplit, как показано ниже, но тогда неясно, как вернуть это обратно в 2 столбца во фрейме данных.

> strsplit(as.character(before$type),'_and_')
[[1]]
[1] "foo" "bar"

[[2]]
[1] "foo"   "bar_2"

[[3]]
[1] "foo" "bar"

[[4]]
[1] "foo"   "bar_2"

Спасибо за любые указатели.Я пока еще не совсем прогнал списки R.

Ответы [ 14 ]

5 голосов
/ 28 сентября 2017

Этот вопрос довольно старый, но я добавлю решение, которое я нашел самым простым на данный момент.

library(reshape2)
before = data.frame(attr = c(1,30,4,6), type=c('foo_and_bar','foo_and_bar_2'))
newColNames <- c("type1", "type2")
newCols <- colsplit(before$type, "_and_", newColNames)
after <- cbind(before, newCols)
after$type <- NULL
after
5 голосов
/ 28 августа 2017

Начиная с версии 3.4.0 R вы можете использовать strcapture() из пакета utils (входит в базовую установку R), привязывая вывод к другим столбцам.

out <- strcapture(
    "(.*)_and_(.*)",
    as.character(before$type),
    data.frame(type_1 = character(), type_2 = character())
)

cbind(before["attr"], out)
#   attr type_1 type_2
# 1    1    foo    bar
# 2   30    foo  bar_2
# 3    4    foo    bar
# 4    6    foo  bar_2
4 голосов
/ 17 февраля 2018

базовый, но, вероятно, медленный:

n <- 1
for(i in strsplit(as.character(before$type),'_and_')){
     before[n, 'type_1'] <- i[[1]]
     before[n, 'type_2'] <- i[[2]]
     n <- n + 1
}

##   attr          type type_1 type_2
## 1    1   foo_and_bar    foo    bar
## 2   30 foo_and_bar_2    foo  bar_2
## 3    4   foo_and_bar    foo    bar
## 4    6 foo_and_bar_2    foo  bar_2
4 голосов
/ 04 декабря 2010

Другой подход, если вы хотите придерживаться strsplit(), это использовать команду unlist(). Вот решение в том же духе.

tmp <- matrix(unlist(strsplit(as.character(before$type), '_and_')), ncol=2,
   byrow=TRUE)
after <- cbind(before$attr, as.data.frame(tmp))
names(after) <- c("attr", "type_1", "type_2")
...