Question

我想按排名更改分类列中的所有值。可以使用列中已排序的唯一元素的索引来确定排名。

例如，

> data[1:5,1] 
[1] "B2" "C4" "C5" "C1" "B5"

然后我希望列中的这些条目替换分类值

> data[1:5,1]  
[1] "1" "4" "5" "3" "2"

另一栏：

> data[1:5,3]
[1] "Verified"        "Source Verified" "Not Verified"    "Source Verified" "Source Verified"

然后是更新后的专栏：

> data[1:5,3]
[1] "3" "2" "1" "2" "2"

我使用此代码执行此任务，但这需要花费很多时间。

for(i in 1:ncol(data)){
  if(is.character(data[,i])){
    temp <- sort(unique(data[,i]))
    for(j in 1:nrow(data)){
      for(k in 1:length(temp)){
        if(data[j,i] == temp[k]){
          data[j,i] <- k}
      }
    }
  }
}

如果可能的话，请建议我这样做的有效方法。感谢。

Answer 1

这是base R中的解决方案。我创建了一个辅助函数，它使用其唯一的排序值作为级别将每个列转换为一个因子。这与您所做的类似，只是我使用as.integer来获取排名值。

rank_fac <- function(col1) 
   as.integer(factor(col1,levels = unique(col1)))

一些数据示例：

dx <- data.frame(
  col1= c("B2" ,"C4" ,"C5", "C1", "B5"),
  col2=c("Verified"    ,    "Source Verified", "Not Verified"  ,  "Source Verified", "Source Verified")
)

在不使用for循环的情况下应用它。最好在这里使用lapply以避免副作用。

data.frame(lapply(dx,rank_fac)

结果：

#       col1 col2
# [1,]    1    3
# [2,]    4    2
# [3,]    5    1
# [4,]    3    2
# [5,]    2    2

使用data.table syntax-sugar

library(data.table)
setDT(dx)[,lapply(.SD,rank_fac)]
#    col1 col2
# 1:    1    3
# 2:    4    2
# 3:    5    1
# 4:    3    2
# 5:    2    2

更简单的解决方案：

仅使用as.integer：

setDT(dx)[,lapply(.SD,as.integer)]

Answer 2

使用match：

# df is your data.frame    
df[] <- lapply(df, function(x) match(x, sort(unique(x))))

在R中将值从分类更改为标称值

2 个答案:

使用data.table syntax-sugar

更简单的解决方案：