python - 根据两列进行排序并将值分配给新列

Question

我有一个数据框作为df(使用pandas模块创建):

     ID   Distance
 0   105   86.7
 1   105   60.5
 2   106   48.3
 3   106   14.1
 4   106   37.2
 5   106   30.8

我正在尝试创建一个列(“Score”)来表示基于相同“ID”的最小距离的分数。

换句话说，对于具有相同“ID”的对象，应用以下规则；

如果对象具有最小距离，“Score” 将得到值“4”。

如果对象具有第二个最小距离，“Score” 将获得值“2”。

如果对象具有第三个最小距离，“Score” 将获得值“1”。

如果对象有第 4 个和第 n 个。最小距离，“Score” 将得到值“0”。

这是我期望的Out_df:

     ID   Distance  Score
 0   105   86.7     2
 1   105   60.5     4
 2   106   48.3     0
 3   106   14.1     4
 4   106   37.2     1
 5   106   30.8     2

我需要帮助来排序问题。

Answer 1

你可以使用

1. groupby获取每个ID
的子DataFrame
2. nsmallest识别这些阈值元素。
3. transform获取每个子数据帧并获得分数。

代码:

def my_sort(s):
    d = dict(zip(s.nsmallest(3), [4, 2, 1]))
    return [d.get(v, 0) for v in s]

df['Score'] = df.groupby('ID').transform(my_sort)

查看 pandas groupby docs有用的 groupby 习语。