소스 검색

Merge branch 'dev' into pro

qingfeng 2 년 전
부모
커밋
d6ba38a9ae
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()