列的信号条件

Question

从 yahoo Finance 下载 ohlcv for nvidia 读取，我正在为信号购买/dontbuy 创建一个列，当我尝试定义哪些通过 avg>volume 测试时，一切要么全部“购买”，要么不购买。df=pd.read_csv('NVDA.csv',dtype={'label':str})df['Price%delta']=((df['Close']/df['Open'])*100)                       df['Avg_volume']=df['Volume'].rolling(7).mean()df['Signal']=0for index, row in df.iterrows():    if row['Volume'] > row['Avg_volume']:    df['Signal']='Buy'    else:        df['Signal']='Dont Buy'

Answer 1

你根本不需要 for 循环：

mask = df["Volume"] > df["Avg_volume"] 

df.loc[mask, "Signal"] = "Buy"

df.loc[~mask, "Signal"] = 'Don't buy'

Answer 2

使用np.where()以下方法的矢量化解决方案：

df['Signal'] = np.where(df['Volume'] > df['Avg_volume'], 'Buy', 'Dont Buy')

Answer 3

您没有指定任何索引来分配'Buy'或'Don't buy'。使用loc来代替：

for index, row in df.iterrows(): 

    if row['Volume'] > row['Avg_volume']:

        df.loc[index, 'Signal']='Buy'

    else:

        df.loc[index, 'Signal']='Dont Buy'