python 根据id这一列新增一列，相同的id命名为一个数字，从0开始

Question

python编程 根据id这一列新增一列，相同的id命名为一个数字，从0开始

Answer 1

import pandas as pd

df = pd.read_excel('data.xlsx')

df.insert(1,'rank',0)

kv={}
def setValue(v):
    if v not in kv:
        kv[v]=len(kv.keys())+1
    return kv[v]

df['rank']=df['id'].apply(setValue)

df.to_excel('result.xlsx', index=False)