创建一个值为0,1,0.5而不是布尔值的列

Question

考虑以下数据帧：

id   Name   Score   
314  John    100    
345  Sara    200
355  Zack    200
333  Harry    50
334  Chad     50
331  Newton  100

我想根据分数为一个新的操作符列指定自定义值，因此如果一个分数小于下一个分数，则它是1，如果它大于0，如果它保持不变，则它是0.5。这就是我想要的样子：

id   Name   Score  Operator
314  John    100      1
345  Sara    200     0.5
355  Zack    200      0
333  Harry    50     0.5
334  Chad     50      1    
331  Newton  100     NAN

我尝试过差分列和布尔列的组合，但它不提供任何摆脱二进制方法的途径

Answer 1

首先，设置您的条件：

prev = df.Score.shift(-1)
c1, c2, c3 = df.Score.lt(prev), df.Score.eq(prev), df.Score.gt(prev)

现在使用numpy.select：

out = df.assign(out=np.select([c1, c2, c3], [1, 0.5, 0], np.nan))

    id    Name  Score  out
0  314    John    100  1.0
1  345    Sara    200  0.5
2  355    Zack    200  0.0
3  333   Harry     50  0.5
4  334    Chad     50  1.0
5  331  Newton    100  NaN

---

下面是另一个仅仅是为了好玩的解决方案（只有当您的差异不小于0.5时才有效）：

df.Score.diff(-1).mul(-1).add(0.5).clip(0, 1)

0    1.0
1    0.5
2    0.0
3    0.5
4    1.0
5    NaN
Name: Score, dtype: float64