迭代数据帧以购买或拒绝股票

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

我试着比较今天和前一天的价格（如果股价比昨天低，“买它”，否则“不要买它”）。我正在尝试使用“shift”功能来检查差异

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

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

Answer 1

我想你可以用diff和np.where

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