Панды получают 1 ранг от группы по нескольким столбцам - PullRequest
Новая
       11

Панды получают 1 ранг от группы по нескольким столбцам

0 голосов
/ 22 февраля 2019

Можно ли сделать что-то подобное 

df = pd.DataFrame({
    "sort_by": ["a","a","a","a","b","b","b", "a"],
     "x": [100.5,200,200,500,1,2,3, 200],
     "y": [4000,2000,2000,1000,500.5,600.5,600.5, 100.5]
})
df = df.sort_values(by=["x","y"], ascending=False)

, где я могу отсортировать по столбцу sort_by и использовать x и y, чтобы найти ранг (используя y, чтобы разорвать связи)

таким идеальным прогнозом будет 

sort_by  x         y       rank
a        500       1000    1
a        200       2000    2
a        200       2000    2
a        200       100.5   3
a        100.5     4000    4
b        3         600.5   1
b        2         600.5   2
b        1         500.5   3

1 Ответ

0 голосов
/ 22 февраля 2019

Проверьте с factorize после sort_values

df = df.sort_values(by=["x","y"], ascending=False)
df['rank']=tuple(zip(df.x,df.y))
df['rank']=df.groupby('sort_by',sort=False)['rank'].apply(lambda x : pd.Series(pd.factorize(x)[0])).values
df
Out[615]: 
  sort_by      x       y  rank
3       a  500.0  1000.0     1
1       a  200.0  2000.0     2
2       a  200.0  2000.0     2
7       a  200.0   100.5     3
0       a  100.5  4000.0     4
6       b    3.0   600.5     1
5       b    2.0   600.5     2
4       b    1.0   500.5     3
...