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

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')