使用 Pandas Python 的连续输赢

Question

我可以在下面输入熊猫系列（收盘价的每日系列）来复制该功能吗？它计算每日连胜数。

cp = pd.Series(index=range(1,8+1), name='Closing_Price',
    data=[20.00, 20.50, 20.75, 19.75, 19.50, 19.35, 19.35, 19.40])

1    20.00
2    20.50
3    20.75
4    19.75
5    19.50
6    19.35
7    19.35
8    19.40

解释是：

• 第 2 天的收盘价高于第 1 天的收盘价，因此我们有单日上涨
• 在第 3 天，价格再次收高，因此我们有 连续两天上涨，即 Streak Duration 值为 2。
• 在第 4 天，收盘价下跌，给我们带来单日下跌。 Streak Duration 值为负 (-1)，因为价格变动是下跌，而不是上涨。
• 下降趋势在第 5 天和第 6 天继续，我们的Streak Duration 值反映为-2 和-3。
• 第 7 天收盘价没有变化，因此 Streak Duration 设置为 0，表示既不是上涨也不是下跌。
• 最后，在第 8 天收盘价再次上涨，使 Streak Duration 值回到 1。

我想我需要 cumsum() 和 groupby。类似的东西

def fxn(series):
    x = series.diff()

但我不知道如何继续

Answer 1

这可能是答案吗？向@smci 大声疾呼，感谢他们提供了急需和备受赞赏的提示！！

def streak(x):
  c = x.copy()
  c['diff'] = c.diff()
  c['sign'] = c['diff'].apply(func = lambda y: y if not y else y // abs(y))
  grouper = (c['sign'] != c['sign'].shift()).cumsum()
  c['streak'] = c['sign'].groupby(grouper).cumsum()
  return c['streak']