遍历数据框以购买或拒绝股票

Question

我为（1年的股价）创建了以下虚拟数据

files

以下是数据帧的开头。

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

我如何比较当日价格与前一天的价格，并创建一列作为“决定”来购买/不购买股票。

Answer 1

我认为您可以使用diff和np.where

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