python - 当另一列增加/减少时添加状态列

Question

当另一列增加/减少或保持不变时，我想在数据框中添加一列: 1 -> 增加，0 -> 相同，-1 -> 减少

所以如果df['battery'] = [1,2,3,4,7,9,3,3,3,]我希望状态为 df['state'] = [1,1,1,1,1,-1,0,0]

Answer 1

这应该可以解决问题!

a = [1,2,3,4,7,9,3,3,3]
b = []

for x in range(len(a)-1):
  b.append((a[x+1] > a[x]) - (a[x+1] < a[x]))

print(b)