| 123456789101112131415161718192021222324252627282930313233343536373839404142 |
- import multiprocessing
- from os import *
- from etl.wind_power.min_sec.ClassIdentifier import ClassIdentifier
- from utils.draw.draw_file import scatter
- from utils.file.trans_methods import read_file_to_df
- def draw(file, fengchang='测试'):
- name = path.basename(file).split('.')[0]
- df = read_file_to_df(file)
- del df['lab']
- identifier = ClassIdentifier(wind_turbine_number='test', origin_df=df, rated_power=5000, cut_out_speed=20,
- active_power='active_power', wind_velocity='wind_velocity',
- pitch_angle_blade='pitch_angle_blade_1')
- df = identifier.run()
- df.loc[df['active_power'] <= 0, 'lab'] = -1
- print(df.groupby('lab').count())
- color_map = {-1: 'red', 0: 'green', 1: 'blue', 2: 'black', 3: 'orange', 4: 'magenta'}
- c = df['lab'].map(color_map)
- # -1:停机 0:好点 1:欠发功率点；2:超发功率点；3:额定风速以上的超发功率点 4: 限电
- legend_map = {"停机": 'red', "好点": 'green', "欠发": 'blue', "超发": 'black', "额定风速以上的超发": 'orange',
- "限电": 'magenta'}
- scatter(name, x_label='风速', y_label='有功功率', x_values=df['wind_velocity'].values,
- y_values=df['active_power'].values, color=c, col_map=legend_map,
- save_file_path=path.dirname(
- path.dirname(__file__)) + sep + "tmp" + sep + str(fengchang) + sep + name + '结果.png')
- if __name__ == '__main__':
- read_dir = r"D:\data\logs\matlib-test"
- files = [read_dir + sep + i for i in listdir(read_dir)]
- if len(files) == 1:
- draw(files[0], "和风元宝山4")
- else:
- with multiprocessing.Pool(4) as pool:
- pool.starmap(draw, [(file, "和风元宝山4") for file in files])
|
