Ver Fonte

合并表数据问题修改

wzl há 6 meses atrás
pai
commit
1c282727f6
1 ficheiros alterados com 4 adições e 3 exclusões
  1. 4 3
      etl/wind_power/min_sec/ReadAndSaveTmp.py

+ 4 - 3
etl/wind_power/min_sec/ReadAndSaveTmp.py

@@ -66,11 +66,12 @@ class ReadAndSaveTmp(object):
                     contains_name = False
                     self.exist_wind_names.append(exist_name)
                 save_path = path.join(merge_path, csv_name)
+                now_df = df[df['wind_turbine_number'] == name]
                 if contains_name:
-                    df.to_csv(save_path, index=False, encoding='utf-8', mode='a',
-                              header=False)
+                    now_df.to_csv(save_path, index=False, encoding='utf-8', mode='a',
+                                  header=False)
                 else:
-                    df.to_csv(save_path, index=False, encoding='utf-8')
+                    now_df.to_csv(save_path, index=False, encoding='utf-8')
 
     def df_save_to_tmp_file(self, df=pd.DataFrame()):