几种写因子的方法

函数

def f(x):
    if 3<=x<=8:
        return 1
    elif x>8:
        return -1
    elif x<3:
        return None
    
temp = pd.DataFrame(self.oo[di,:], columns=['oo'])
temp['alpha'] = temp.oo.apply(lambda x: f(x))
self.alpha[v] = temp.alpha[v]

直接对数组进行操作

def generate(self, didx):
    di = didx - self.delay
    self.getRefreshData()
    v = self.valid[di]
    
    temp = self.oo[di, :].copy()
    temp[np.isnan(temp)] = 0
    temp[temp <= 3] = 0
    temp[temp >= 8] = -1
    temp[(temp < 8) & (temp > 3)] = 1
    temp[temp == 0] = np.nan
    self.alpha[v] = temp[v]
def serialize(self):
    pass