powerScatter2DAnalyst.py 5.9 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117
  1. import os
  2. from datetime import datetime
  3. import pandas as pd
  4. from algorithmContract.confBusiness import *
  5. from algorithmContract.contract import Contract
  6. from behavior.analystWithGoodBadPoint import AnalystWithGoodBadPoint
  7. class PowerScatter2DAnalyst(AnalystWithGoodBadPoint):
  8. """
  9. 风电机组功率曲线散点分析。
  10. 秒级scada数据运算太慢,建议使用分钟级scada数据
  11. """
  12. def typeAnalyst(self):
  13. return "power_scatter_2D"
  14. def selectColumns(self):
  15. return [Field_DeviceCode, Field_Time, Field_WindSpeed, Field_ActiverPower]
  16. def addPropertyToDataFrame(self,dataFrameOfTurbine : pd.DataFrame, currTurbineInfo : pd.Series, currTurbineModelInfo : pd.Series):
  17. dataFrameOfTurbine[Field_PowerFarmCode] = self.currPowerFarmInfo[Field_PowerFarmCode]
  18. dataFrameOfTurbine[Field_MillTypeCode] = currTurbineModelInfo[Field_MillTypeCode]
  19. def turbinesAnalysis(self, outputAnalysisDir, conf: Contract, turbineCodes):
  20. dictionary = self.processTurbineData(turbineCodes, conf, self.selectColumns())
  21. dataFrame = self.userDataFrame(dictionary, conf.dataContract.configAnalysis, self)
  22. turbineInfos = self.common.getTurbineInfos(conf.dataContract.dataFilter.powerFarmID, turbineCodes, self.turbineInfo)
  23. if len(dataFrame) <= 0:
  24. self.logger.info("After screening for blade pitch angle less than the configured value, plot power curve scatter points without data")
  25. return
  26. return self.drawOfPowerCurveScatter(dataFrame, turbineInfos,outputAnalysisDir, conf, self.dataFrameContractOfTurbine)
  27. def drawOfPowerCurveScatter(self, dataFrame: pd.DataFrame, turbineModelInfo: pd.Series, outputAnalysisDir, conf: Contract, dataFrameGuaranteePowerCurve: pd.DataFrame):
  28. """
  29. 绘制风速-功率分布图并保存为文件。
  30. 参数:
  31. dataFrameMerge (pd.DataFrame): 包含数据的DataFrame,需要包含设备名、风速和功率列。
  32. csvPowerCurveFilePath (str): 功率曲线文件路径。
  33. outputAnalysisDir (str): 分析输出目录。
  34. confData (ConfBusiness): 配置
  35. """
  36. #机型切入风速 series
  37. cutInWsField = self.turbineModelInfo[Field_CutInWS]
  38. cut_in_ws = cutInWsField.min() - 1 if cutInWsField.notna().any() else 2
  39. # if not dataFrame.empty and Field_CutInWS in dataFrame.columns and dataFrame[Field_CutInWS].notna().any():
  40. # cut_in_ws = dataFrame[Field_CutInWS].min() - 1
  41. # else:
  42. # cut_in_ws = 2
  43. # 按设备名分组数据
  44. grouped = dataFrame.groupby([Field_NameOfTurbine, Field_CodeOfTurbine])
  45. result_rows = []
  46. # 遍历每个设备的数据
  47. for name, group in grouped:
  48. #获取当前风机信息dataFrame
  49. currentEngineDataFrame = turbineModelInfo[turbineModelInfo[Field_CodeOfTurbine]==name[1]]
  50. #获取当前机型
  51. millTypeCode = currentEngineDataFrame.get(Field_MillTypeCode, "").iloc[0]
  52. #当前机型合同功率曲线dataFrame
  53. currentMillTypePowerDataFrame = dataFrameGuaranteePowerCurve[dataFrameGuaranteePowerCurve[Field_MillTypeCode] == millTypeCode]
  54. # 获取机型的名字(machine_type_code)
  55. engineTypeName = self.common.getTurbineModelByCode(millTypeCode, self.turbineModelInfo)[Field_MachineTypeCode]
  56. # 使用 apply() 对每个元素调用 datetime.fromtimestamp
  57. group['monthIntTime'] = group['monthIntTime'].apply(lambda x: datetime.fromtimestamp(x).strftime('%Y-%m'))
  58. # 定义要替换的空值类型
  59. na_values = {pd.NA, float('nan')}
  60. # 构建最终的JSON对象
  61. json_output = {
  62. "analysisTypeCode": "逐月有功功率散点2D分析",
  63. "engineCode": millTypeCode,
  64. "engineTypeName": engineTypeName,
  65. "xaixs": "风速(m/s)",
  66. "yaixs": "有功功率(kW)",
  67. "data": [
  68. {# 提取机组数据
  69. "engineName": name[0],
  70. "engineCode": name[1],
  71. "title":f' 逐月有功功率散点2D分析-机组: {name[0]}',
  72. "xData": group[Field_WindSpeed].replace(na_values, None).tolist(),
  73. "xrange":[cut_in_ws, 25],
  74. "yData": group[Field_ActiverPower].replace(na_values, None).tolist(),
  75. "yrange":[self.axisLowerLimitActivePower,self.axisUpperLimitActivePower],
  76. "colorbar": group['monthIntTime'].tolist(),
  77. "colorbartitle": "年月",
  78. "mode":"markers"
  79. },
  80. {# 提取合同功率曲线数据
  81. "enginName": "合同功率曲线",
  82. "xData":currentMillTypePowerDataFrame[Field_WindSpeed].replace(na_values, None).tolist(),
  83. "yData":currentMillTypePowerDataFrame[Field_ActiverPower].replace(na_values, None).tolist(),
  84. "zData": [],
  85. "mode":"lines+markers"
  86. }]
  87. }
  88. # 将JSON对象保存到文件
  89. output_json_path = os.path.join(outputAnalysisDir, f"{name[0]}-scatter.json")
  90. with open(output_json_path, 'w', encoding='utf-8') as f:
  91. import json
  92. json.dump(json_output, f, ensure_ascii=False, indent=4)
  93. # 如果需要返回DataFrame,可以包含文件路径
  94. result_rows.append({
  95. Field_Return_TypeAnalyst: self.typeAnalyst(),
  96. Field_PowerFarmCode: conf.dataContract.dataFilter.powerFarmID,
  97. Field_Return_BatchCode: conf.dataContract.dataFilter.dataBatchNum,
  98. Field_CodeOfTurbine: name[1],
  99. Field_Return_FilePath: output_json_path,
  100. Field_Return_IsSaveDatabase: True
  101. })
  102. result_df = pd.DataFrame(result_rows)
  103. return result_df