Укажите порядок столбцов при приведении длинных к широким - PullRequest
Новая
       16

Укажите порядок столбцов при приведении длинных к широким

1 голос
/ 03 июня 2019

Я хочу привести таблицу к широкоформатному формату, используя reshape2. Я хочу, чтобы столбцы были упорядочены по переменным, т.е. следующим образом: 

data <- as.data.frame(matrix(c(rep(1,3),rep(2,3),rep(1:3,2),rep(0:2,4)),6,4))
colnames(data) <- c("id", "rater", "x", "y")
print(data)

#   id rater x y
# 1  1     1 0 0
# 2  1     2 1 1
# 3  1     3 2 2
# 4  2     1 0 0
# 5  2     2 1 1
# 6  2     3 2 2

Для использования в качестве такового: 

# Result:
#   id x.1 y.1 x.2 y.2 x.3 y.3
# 1  1   0   0   1   1   2   2
# 4  2   0   0   1   1   2   2

x, за которым следует y для каждого образца.

Сейчас я использую dcast и получаю следующий вывод: 

dcast(as.data.table(data), id~rater, value.var=c("x", "y"), sep=".")
#   id x.1 x.2 x.3 y.1 y.2 y.3
#1:  1   0   1   2   0   1   2
#2:  2   0   1   2   0   1   2

Но я хочу, чтобы это было x.1, y.1, x.2, y.2 и т. Д.

Я могу сделать это с помощью изменения формы (оригинал), но это занимает слишком много времени для моих данных (более 500 000 строк, 15 минут + на таблицу, плюс 20 + ГБ памяти) 

reshape(data, idvar = id, timevar = "rater", direction = "wide")

Спасибо!

1 Ответ

2 голосов
/ 03 июня 2019

Опция заключается в извлечении числовой части имен столбцов, order

out <- dcast(as.data.table(data), id~rater, value.var=c("x", "y"), sep=".")
setcolorder(out,  c(1, order(as.numeric(gsub("\\D+", "", names(out)[-1])))+1))
out
#   id x.1 y.1 x.2 y.2 x.3 y.3
#1:  1   0   0   1   1   2   2
#2:  2   0   0   1   1   2   2
...