Pandas dày đặc RANK

Question

Tôi đang đối phó với gấu trúc dataframe và có một khung như thế này:

Year Value 
2012 10 
2013 20 
2013 25 
2014 30 

Tôi muốn làm một equialent để DENSE_RANK() trên (thứ tự theo năm) chức năng. để tạo cột bổ sung như thế này:

Year Value Rank 
    2012 10 1 
    2013 20 2 
    2013 25 2 
    2014 30 3 

Làm thế nào để có thể thực hiện được trong gấu trúc?

Cảm ơn!

Answer 1

Sử dụng pd.Series.rank với method='dense'

df['Rank'] = df.Year.rank(method='dense').astype(int) 

df 

Answer 2

Bạn có thể chuyển đổi năm thành phân loại và sau đó lấy mã của chúng (thêm một vì chúng không được lập chỉ mục và bạn muốn giá trị ban đầu bắt đầu bằng một ví dụ của bạn).

df['Rank'] = df.Year.astype('category').cat.codes + 1 

>>> df 
    Year Value Rank 
0 2012  10  1 
1 2013  20  2 
2 2013  25  2 
3 2014  30  3 

Answer 3

Giải pháp nhanh nhất là factorize:

df['Rank'] = pd.factorize(df.Year)[0] + 1 

Thời gian:

#len(df)=40k 
df = pd.concat([df]*10000).reset_index(drop=True) 

In [13]: %timeit df['Rank'] = df.Year.rank(method='dense').astype(int) 
1000 loops, best of 3: 1.55 ms per loop 

In [14]: %timeit df['Rank1'] = df.Year.astype('category').cat.codes + 1 
1000 loops, best of 3: 1.22 ms per loop 

In [15]: %timeit df['Rank2'] = pd.factorize(df.Year)[0] + 1 
1000 loops, best of 3: 737 µs per loop