Pandas 列基于上一行多列标准

Question

我知道这已被问过几次，但我无法解决它。 我非常抱歉。 我有一个 dataframe 'stock_df' ，其中 1 列具有 'RSI_14' 值。 我是新手，所以为了更容易，我用 == 0 为 3 个状态初始化 3 列。 这个辅助栏目是为了更好地解释这个问题：

• 超过 70 --> 'plus70' = 1
• 小于 30 --> 'minus30' = 1
• 在 70 到 30 之间 --> '介于' = 1

目标是在 ' RSI_14 ' > 70 和-1当 'RSI_14' < 30（这很容易）时产生一个“信号” ，但棘手的部分是当 state 是“介于”70 和 30 之间时需要将前+1 或 -1 state 放在'Signal'中，并保持该数字直到 state 的下一个 +1 或 -1变化，并一直保持下去......这应该不难。 shift(1) 或 .diff(1)，但我不明白。

这是期望的结果：结果

我试过np.where，但它是最后一个似乎不起作用的“stock_df['Signal'].shift(1)”：

stock_df['Signal'] = np.where(stock_df['RSI_14'] > 70, 1, (np.where(stock_df['RSI_14'] < 30, -1, stock_df['Signal'].shift(1))))

我认为解决方案必须在“groupby”和“transform”中，但我尝试了很多不同的方法，但我很笨拙......我真的认为是 groupby。 我在这里检查了很多答案，但我无法解决它。 我真的很感谢你的帮助。 谢谢

Answer 1

我认为您不需要创建任何其他列来实现此目的。 只需使用：

condition_values = stock_df.RSI_14.values
signal = []
last = None
for item in condition_values:
    if item < 30:
        signal.append(-1)
        last = -1
    elif item > 70:
        signal.append(1)
        last = 1
    else:
        signal.append(last)

 df['signal'] = signal

将开头的 none 更改为您喜欢的值。

Answer 2

以下代码片段可以工作。 将信号值分配为等于前一行信号值（之间等于 1 时）应该可以完成工作。

for i in range(len(stock_df)):
    if stock_df['between'][i] == 1:
        if i == 0:
            stock_df['Signal'][i] = 1
        else:
            stock_df['Signal'][i] = stock_df['Signal'][i - 1]