找到条件中的第一个信号[重复]

Question

我有一个具有不同“价格”的数据框 (df)，我想比较这些价格并做出决定。

df['Decision'] = np.where((df['price1'] > df['price2']) ,'sell',np.where((df['price1'] < df['price2']),'buy',np.nan))

我的输出是：

price1	price2	Decision
50	50	NaN
100	200	buy
70	140	buy
150	200	buy
150	50	sell
60	20	sell

但我只想拥有“买入”或“卖出”的“第一个信号”并删除复制直到下一个信号，如：

price1	price2	Decision
50	50	NaN
100	200	buy
70	140
150	200
150	50	sell
60	20

Answer 1

你可以试试idxmax:

df.loc[(df['price1'] > df['price2']).idxmax(), 'Decision'] = 'sell'
df.loc[(df['price1'] < df['price2']).idxmax(), 'Decision'] = 'buy'