Browse Source

Merge branch 'dev_david' of anweiguo/algorithm_platform into dev_awg

liudawei 5 tháng trước cách đây
mục cha
commit
7fe6afa95b
1 tập tin đã thay đổi với 1 bổ sung1 xóa
  1. 1 1
      data_processing/data_operation/pre_data_ftp.py

+ 1 - 1
data_processing/data_operation/pre_data_ftp.py

@@ -176,7 +176,7 @@ def get_nwp_from_ftp():
         df['date_time'] = pd.to_datetime(df['date_time'])
         dfs = df.groupby('farm_id')
         for farm_id, df in dfs:
-            if df.columns.isin(['cdq', 'dq', 'zq']):
+            if {'cdq', 'dq', 'zq'}.issubset(set(df.columns.tolist())):
                 cdq, dq, zq = df.loc[df.index[0], ['cdq', 'dq', 'zq']]
             else:
                 cdq, dq, zq = 0, 0, 0