read_and_draw_png.py 1.6 KB

1234567891011121314151617181920212223242526272829303132333435363738
  1. import multiprocessing
  2. import os
  3. from etl.wind_power.min_sec.ClassIdentifier import ClassIdentifier
  4. from utils.draw.draw_file import scatter
  5. from utils.file.trans_methods import read_file_to_df
  6. def draw(file, fengchang='测试'):
  7. name = os.path.basename(file).split('.')[0]
  8. df = read_file_to_df(file)
  9. identifier = ClassIdentifier(wind_turbine_number='test', origin_df=df, rated_power=1500, cut_out_speed=25,active_power='功率',wind_velocity='风速')
  10. df = identifier.run()
  11. df.loc[df['active_power'] <= 0, 'lab'] = -1
  12. color_map = {-1: 'red', 0: 'green', 1: 'blue', 2: 'black', 3: 'orange', 4: 'magenta'}
  13. c = df['lab'].map(color_map)
  14. # -1:停机 0:好点 1:欠发功率点;2:超发功率点;3:额定风速以上的超发功率点 4: 限电
  15. legend_map = {"停机": 'red', "好点": 'green', "欠发": 'blue', "超发": 'black', "额定风速以上的超发": 'orange', "限电": 'magenta'}
  16. scatter(name, x_label='风速', y_label='有功功率', x_values=df['wind_velocity'].values,
  17. y_values=df['active_power'].values, color=c, col_map=legend_map,
  18. save_file_path=os.path.dirname(
  19. os.path.dirname(__file__)) + os.sep + "tmp" + os.sep + str(fengchang) + os.sep + name + '结果.png')
  20. if __name__ == '__main__':
  21. read_dir = r"D:\data\清理数据\和风元宝山\test_11_test\test_A01"
  22. files = [read_dir + os.sep + i for i in os.listdir(read_dir)]
  23. if len(files) == 1:
  24. draw(files[0], "和风元宝山4")
  25. else:
  26. with multiprocessing.Pool(4) as pool:
  27. pool.starmap(draw, [(file, "和风元宝山4") for file in files])