浏览代码

合并表数据问题修改

wzl 6 月之前
父节点
当前提交
1c282727f6
共有 1 个文件被更改,包括 4 次插入3 次删除
  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()):