Thêm một cột của bậc

Question

Tôi có một số dữ liệu:

test <- data.frame(A=c("aaabbb", 
"aaaabb", 
"aaaabb", 
"aaaaab", 
"bbbaaa") 
) 

và vân vân. Tất cả các phần tử đều có cùng độ dài và đã được sắp xếp trước khi tôi nhận được chúng.

Tôi cần tạo cột xếp hạng mới, "Thứ nhất", "Thứ hai", "Thứ ba", bất kỳ thứ gì sau đó có thể để trống và cần phải tính đến các mối quan hệ. Vì vậy, trong trường hợp trên, tôi muốn có được kết quả như sau:

A  B 
aaabbb First 
aaaabb Second 
aaaabb Second 
aaaaab Third 
bbbaaa 
bbbbaa 

Tôi đã xem thứ hạng() và một số bài đăng khác đã sử dụng nó, nhưng tôi không thể làm điều đó đang được tìm kiếm.

Answer 1

Làm thế nào về điều này:

test$B <- match(test$A , unique(test$A)[1:3]) 
test 
     A B 
1 aaabbb 1 
2 aaaabb 2 
3 aaaabb 2 
4 aaaaab 3 
5 bbbaaa NA 
6 bbbbaa NA 

Một trong nhiều cách để làm điều này. Có thể không phải là tốt nhất, nhưng có thể dễ dàng ghi nhớ và khá trực quan. Bạn có thể sử dụng unique vì bạn nhận được dữ liệu được sắp xếp trước.

Khi dữ liệu được sắp xếp một chức năng phù hợp đáng xem xét là rle, mặc dù nó hơi tù hơn trong ví dụ này:

rnk <- rle(as.integer(df$A))$lengths 
rnk 
# [1] 1 2 1 1 1 
test$B <- c(rep(1:3 , times = rnk[1:3]) , rep(NA, sum(rnk[-c(1:3)]))) 

rle tính độ dài (và các giá trị mà chúng tôi chưa thực sự quan tâm ở đây) của chạy các giá trị bằng nhau trong một vec-tơ - vì vậy một lần nữa điều này hoạt động vì dữ liệu của bạn đã được sắp xếp.

Và nếu bạn không có để có khoảng trống sau khi mặt hàng đứng thứ ba nó thậm chí còn đơn giản hơn (và dễ đọc hơn):

test$B <- rep(1:length(rnk),times=rnk) 

Answer 2

Điều này có vẻ giống như một ứng dụng tốt cho các yếu tố:

test$B <- as.numeric(factor(test$A, levels = unique(test$A))) 

cumsum cũng đề cập đến nơi chúng tôi thêm 1 mỗi khi giá trị thay đổi:

test$B <- cumsum(c(TRUE, tail(test$A, -1) != head(test$A, -1))) 

(Giống như @Simon đã nói, có nhiều cách để thực hiện việc này ...)