按顺序获取每一行的列名

时间:2016-11-10 17:39:51

标签: r performance apply

我有一个包含三列数值的数据框。我想对每一行进行排序,并按照每行的排序顺序打印一个包含列名的字符串。这是我的代码:

> df <- data.frame(x = c(1,2,3), y = c(3,1,2), d = c(4,0, 5))
> df
  x y d
1 1 3 4
2 2 1 0
3 3 2 5
> for(r in 1:nrow(df))
+ print(paste(colnames(sort(df[r,])), collapse = " "))
[1] "x y d"
[1] "d y x"
[1] "y x d"

这确实有效,但是当我拥有大型数据帧时需要很长时间。有没有更有效的方法来执行排序?我尝试使用apply,但这只是给了空字符串:

> apply(df, 1, function(row) paste( colnames( sort( row, decreasing = T )) , collapse = " " ))
[1] "" "" ""

当我在单行中调用apply中的函数时,它可以工作:

> paste( colnames( sort( df[1,], decreasing = T )) , sep = " " )
[1] "d" "y" "x"

我主要是在寻找一种更快的方法来在大数据帧的每一行上执行此操作而不是for循环。我认为应用可能会更快,但无法让它发挥作用。

1 个答案:

答案 0 :(得分:6)

t(apply(df, 1, function(x) names(x)[order(x)]))

这个问题在我的阅读中再次出现,所以我想应该编辑以添加更多的方法。以后可能会有所帮助:

library(data.table)
setDT(df)[, paste(colnames(df)[order(.SD)], collapse = " "), by = 1:nrow(df)]

逻辑:groupby行索引(表示逐行操作) - 然后将rank应用于每个组(基本上是一行)。 .SD表示数据子集(所有列)(您也可以使用.SDcols =参数控制该数据。然后简单paste相应的列名称

dplyr

中实现的逻辑相同
library(dplyr)
library(tidyr)
df %>% rowwise() %>% 
       do(rank = paste(colnames(df)[order(unlist(.))], collapse = " ")) %>% 
       unnest()

输出:

#   nrow    V1
#1:    1 x y d
#2:    2 d y x
#3:    3 y x d