Итерация по фрейму данных для покупки или отклонения акций

Question

Я создал следующие фиктивные данные для (1 год цен на акции)

df = pd.DataFrame(np.random.uniform(100,200, size = (365,1)), index=list_dates, columns=list('A'))
df.head()

Ниже приводится заголовок фрейма данных.

                A
2018-01-01  151.034052
2018-01-02  157.682393
2018-01-03  134.194843
2018-01-04  199.916593
2018-01-05  194.399690

Я пытаюсь сравнить цену сегодня и позавчера (если цены на акции меньше, чем вчера, «купить», иначе «не покупать». Я пытаюсь использовать функцию «сдвиг», чтобы проверить разница.

df['difference'] = df['A'] - df['A'].shift(1)

Как сравнить цену текущего дня с днем ​​ранее и создать столбец как «решение» покупать / не покупать акции.

Answer 1

Я думаю, вы можете использовать diff и np.where

df['decision'] = np.where(df.A.diff().ge(0), 'buy', 'sell')