2 次代码提交 4c11fb1251 ... d6ba38a9ae

作者 SHA1 备注 提交日期
  qingfeng d6ba38a9ae Merge branch 'dev' into pro 2 年之前
  zhuxi 4527bdf677 faultclass 2 年之前
共有 1 个文件被更改,包括 8 次插入0 次删除
  1. 8 0
      LIB/MIDDLE/FaultClass/V1_0_0/faultclass.py

+ 8 - 0
LIB/MIDDLE/FaultClass/V1_0_0/faultclass.py

@@ -176,6 +176,14 @@ def arrange(result,result_final,start_time):
             res_update=res_update.append(result0)
     return res_new,res_update
 
+def arrange2(dataorg,df_res,start_time,fault_name):
+    res_new=df_res.copy()
+    res_update=pd.DataFrame()
+    if len(dataorg)>0:
+        dataorg=dataorg[dataorg['fault_class']==fault_name]
+        res_new,res_update=arrange(df_res,dataorg,start_time)
+    return res_new,res_update
+
 # Step8 Process
 def pred(data_fea,model,scaler,col,end_time,time_steps):
     df_res=pd.DataFrame()