python - 查找条件中的第一个信号

Question

这个问题在这里已经有了答案:





Pandas - Replace Duplicates with Nan and Keep Row

(3 个回答)


3个月前关闭。




我有一个具有不同“价格”的数据框 (df)，我想比较这些价格并做出决定。df['Decision'] = np.where((df['price1'] > df['price2']) ,'sell',np.where((df['price1'] < df['price2']),'buy',np.nan))我的输出是:


价格1
价格2
决定


50
50
NaN

100
200
买

70
140
买

150
200
买

150
50
卖

60
20
卖


但我只想拥有“买入”或“卖出”的“第一个信号”和 删除复制直到下一个信号 ， 作为:


价格1
价格2
决定


50
50
NaN

100
200
买

70
140

150
200

150
50
卖

60
20

Answer 1

你可以试试 idxmax :

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