data_analyse_origin.py 48 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058105910601061106210631064106510661067106810691070107110721073107410751076107710781079108010811082108310841085108610871088108910901091109210931094109510961097109810991100110111021103110411051106110711081109111011111112111311141115111611171118111911201121112211231124112511261127112811291130113111321133113411351136113711381139
  1. import os
  2. import json
  3. import pandas as pd
  4. import numpy as np
  5. import seaborn as sns
  6. import matplotlib.pyplot as plt
  7. from matplotlib.ticker import MaxNLocator
  8. from typing import Tuple, List
  9. import warnings
  10. import time
  11. import sys
  12. import frequency_filter as ff
  13. from datetime import datetime
  14. warnings.filterwarnings("ignore", category=FutureWarning) # 忽略特定警告
  15. plt.rcParams['font.sans-serif'] = ['SimHei'] # 使用黑体
  16. plt.rcParams['axes.unicode_minus'] = False # 解决保存图像是负号'-'显示为方块的问题
  17. def result_main():
  18. """
  19. 创建data目录,返回历史分析数据存放的文件路径
  20. """
  21. # 获取当前程序的绝对路径
  22. python_interpreter_path = sys.executable
  23. project_directory = os.path.dirname(python_interpreter_path)
  24. data_folder = os.path.join(project_directory, 'data')
  25. # 检查data文件夹是否存在,如果不存在则创建
  26. if not os.path.exists(data_folder):
  27. os.makedirs(data_folder)
  28. # CSV文件路径
  29. csv_file_path = os.path.join(data_folder, 'history_data.csv')
  30. # 检查CSV文件是否存在,如果不存在则创建一个空的CSV文件
  31. if not os.path.exists(csv_file_path):
  32. pd.DataFrame(columns=['时间', '场站', '风机编号', '采样频率',
  33. '叶片1角度偏差', '叶片2角度偏差', '叶片3角度偏差', '相对角度偏差',
  34. '叶片1净空值', '叶片2净空值', '叶片3净空值',
  35. '叶片1扭转', '叶片2扭转', '叶片3扭转', '平均扭转',
  36. '振动幅值', '振动主频']).to_csv(csv_file_path, index=False)
  37. return csv_file_path
  38. def delete_data(name):
  39. """
  40. 删除历史分析数据
  41. :param name: 删除条件
  42. :return: csv文件路径
  43. """
  44. # 获取当前程序的绝对路径
  45. python_interpreter_path = sys.executable
  46. project_directory = os.path.dirname(python_interpreter_path)
  47. data_folder = os.path.join(project_directory, 'data')
  48. # CSV文件路径
  49. csv_file_path = os.path.join(data_folder, 'history_data.csv')
  50. df = pd.read_csv(csv_file_path)
  51. condition = ((df['时间'].astype(str).str.contains(name[0])) &
  52. (df['场站'].astype(str).str.contains(name[1])) &
  53. (df['风机编号'].astype(str).str.contains(name[2])))
  54. # 删除满足条件的行
  55. df = df[~condition]
  56. # 如果需要,可以将修改后的 DataFrame 保存回 CSV 文件
  57. df.to_csv(csv_file_path, index=False)
  58. return csv_file_path
  59. def history_data(name):
  60. """
  61. 读取历史分析数据
  62. :param name: 接口返回列表
  63. :return:
  64. """
  65. time_code = name[0]
  66. wind_name = name[1]
  67. turbine_code = name[2]
  68. # 获取当前程序的绝对路径
  69. python_interpreter_path = sys.executable
  70. project_directory = os.path.dirname(python_interpreter_path)
  71. data_folder = os.path.join(project_directory, 'data')
  72. time_code_cleaned = time_code.replace("-", "").replace(":", "").replace(" ", "")
  73. json_filename = f"{wind_name}_{turbine_code}_{time_code_cleaned}.json"
  74. json_file_path = os.path.join(data_folder, json_filename)
  75. if not os.path.exists(json_file_path):
  76. raise ValueError("文件不存在")
  77. with open(json_file_path, 'r') as f:
  78. data = json.load(f)
  79. return data
  80. def data_analyse(path: List[str]):
  81. """
  82. 创建data目录,把分析数据保存到历史记录中,同时返回全量分析数据
  83. """
  84. # 基础配置参数
  85. locate_file = path[0]
  86. measure_file = path[1]
  87. noise_reduction = 0.000001 # 如果一个距离值的所有样本量小于总样本量的noise_reduction,则被去掉
  88. min_difference = 1.5 # 如果相邻2个点的距离差大于min_difference,则被注意是否是周期节点
  89. angle_cone = float(path[2]) # 锥角
  90. axial_inclination = float(path[3]) # 轴向倾角
  91. return_list = []
  92. # 读取文件信息,包括风场名、风机编号、时间、采样频率、2个通道俯仰角
  93. wind_name, turbine_code, time_code, sampling_fq, angle_nan, angle_cen = find_param(locate_file)
  94. wind_name_1, turbine_code_1, time_code_1, sampling_fq_1, angle_tip, angle_root = find_param(measure_file)
  95. sampling_fq_1 = sampling_fq_1 * 1000
  96. sampling_fq = sampling_fq * 1000
  97. print(wind_name, turbine_code, time_code, sampling_fq, angle_nan, angle_cen)
  98. print(wind_name_1, turbine_code_1, time_code, sampling_fq_1, angle_tip, angle_root)
  99. # 读取数据,并检查是否有时间序列异常,分离2通道数据
  100. data_nan, data_cen = process_data(locate_file)
  101. data_tip, data_root = process_data(measure_file)
  102. # 全部数据进行降噪、去除异常点处理,叶根叶尖数据计算叶片扫掠起始、结束点,轮毂中心数据计算距离均值
  103. start_tip, end_tip, filtered_data_tip = cycle_calculate(data_tip, noise_reduction, min_difference)
  104. start_root, end_root, filtered_data_root = cycle_calculate(data_root, noise_reduction, min_difference)
  105. start_nan, end_nan, filtered_data_nan = cycle_calculate(data_nan, noise_reduction, min_difference)
  106. filtered_data_cen = tower_filter(data_cen, noise_reduction)
  107. dist_cen = np.mean(filtered_data_cen.iloc[:, 1].tolist())
  108. filtered_data_cen.iloc[:, 1] = filtered_data_cen.iloc[:, 1] * np.cos(np.deg2rad(angle_cen))
  109. # 检查起始结束点顺序,确保叶根叶尖测点同步开始、结束
  110. if end_tip.iloc[0, 0] < start_root.iloc[0, 0]:
  111. start_tip = start_tip.drop(start_tip.index[0])
  112. end_tip = end_tip.drop(end_tip.index[0])
  113. if start_root.iloc[0, 0] < start_tip.iloc[0, 0] < end_tip.iloc[0, 0] < end_root.iloc[0, 0]:
  114. pass
  115. else:
  116. raise ValueError("The elements are not in the expected order.")
  117. # 计算叶根、叶尖处的塔筒距离,对轮毂中心做FFT分析
  118. tower_dist_tip = ff.tower_cal(filtered_data_tip, start_tip, end_tip, sampling_fq_1)
  119. tower_dist_root = ff.tower_cal(filtered_data_root, start_root, end_root, sampling_fq_1)
  120. tower_dist_nan = ff.tower_cal(filtered_data_nan, start_nan, end_nan, sampling_fq)
  121. lowpass_data, fft_x, fft_y, tower_freq, tower_max= ff.process_fft(filtered_data_cen, sampling_fq)
  122. # 根据起始结束点,对叶根、对叶片数据进行归一化处理,计算每个叶片的散点表、线表、边界点表、标准循环周期长度、每个叶片平均最小值
  123. result_line_tip, result_scatter_tip, border_rows_tip, cycle_len_tip, min_tip \
  124. = data_normalize(filtered_data_tip, start_tip, end_tip)
  125. result_line_root, result_scatter_root, border_rows_root, cycle_len_root, min_root \
  126. = data_normalize(filtered_data_root, start_root, end_root)
  127. result_line_nan, result_scatter_nan, border_rows_nan, cycle_len_nan, min_nan \
  128. = data_normalize(filtered_data_nan, start_nan, end_nan)
  129. # 计算3个叶片的平均轮廓,3个叶片的形状差
  130. result_avg_tip, result_diff_tip = blade_shape(result_line_tip)
  131. result_avg_root, result_diff_root = blade_shape(result_line_root)
  132. # 对叶尖的边界点表和俯仰角做坐标归一化处理
  133. border_rows_tip_new, angle_tip_new = coordinate_normalize(border_rows_tip, angle_tip)
  134. border_rows_nan_new, angle_nan_new = coordinate_normalize(border_rows_nan, angle_nan)
  135. # 对叶片的边界点表做半径计算
  136. tip_r = radius_cal(border_rows_tip_new, angle_tip_new, dist_cen, angle_cen, axial_inclination, angle_cone)
  137. root_r = radius_cal(border_rows_root, angle_root, dist_cen, angle_cen, axial_inclination, angle_cone)
  138. nan_r = radius_cal(border_rows_nan_new, angle_nan_new, dist_cen, angle_cen, axial_inclination, angle_cone)
  139. # 计算叶片测量位置处的绝对桨距角、相对桨距角、线速度、叶片内部中心点距离
  140. pitch_angle_tip, aero_dist_tip, v_speed_tip, cen_blade_tip = (
  141. blade_angle_aero_dist(border_rows_tip, tip_r, cycle_len_tip, tower_dist_tip, angle_tip_new))
  142. pitch_angle_root, aero_dist_root, v_speed_root, cen_blade_root = (
  143. blade_angle_aero_dist(border_rows_root, root_r, cycle_len_root, tower_dist_root, angle_root))
  144. pitch_angle_nan, aero_dist_nan, v_speed_nan, cen_blade_nan = (
  145. blade_angle_aero_dist(border_rows_nan_new, nan_r, cycle_len_nan, tower_dist_nan, angle_nan_new))
  146. # 将列表转换为 numpy 数组
  147. cen_blade_tip_array = np.array(cen_blade_tip)
  148. cen_blade_nan_array = np.array(cen_blade_nan)
  149. min_tip_array = np.array(min_tip)
  150. min_nan_array = np.array(min_nan)
  151. abs_diff = np.abs(cen_blade_tip_array - min_tip_array) # 计算差值的绝对值
  152. abs_diff_nan = np.abs(cen_blade_nan_array - min_nan_array) # 计算差值的绝对值
  153. blade_dist_tip = abs_diff * np.cos(np.deg2rad(angle_tip_new))
  154. blade_dist_nan = abs_diff_nan * np.cos(np.deg2rad(angle_nan_new))
  155. blade_dist_tip.tolist() # 如果需要将结果转换回列表
  156. blade_dist_nan.tolist() # 如果需要将结果转换回列表
  157. # 计算叶片转速-净空散点表
  158. dist_distribute = blade_dist_distribute_cal(filtered_data_tip, start_tip, end_tip,
  159. tower_dist_tip, angle_tip_new, blade_dist_tip)
  160. dist_distribute_nan = blade_dist_distribute_cal(filtered_data_nan, start_nan, end_nan,
  161. tower_dist_nan, angle_nan_new, blade_dist_nan)
  162. # dist_distribute = [df.round(5) for df in dist_distribute]
  163. dist_distribute = [df.round(5) for df in dist_distribute_nan]
  164. # 获取净空距离的最小值和最大值,以及它们对应的转速值,并分别保存在列表中
  165. min_values = []
  166. min_keys = []
  167. max_values = []
  168. max_keys = []
  169. mean_values = []
  170. for df in dist_distribute:
  171. second_col_min = df[df.columns[1]].min()
  172. second_col_max = df[df.columns[1]].max()
  173. min_row = df[df[df.columns[1]] == second_col_min]
  174. max_row = df[df[df.columns[1]] == second_col_max]
  175. min_values.append(round(second_col_min, 2))
  176. min_keys.append(round(min_row.iloc[0][df.columns[0]], 2))
  177. max_values.append(round(second_col_max, 2))
  178. max_keys.append(round(max_row.iloc[0][df.columns[0]], 2))
  179. for i in range(3):
  180. mean_values.append(round((max_values[i] + min_values[i]) / 2, 2))
  181. # 将叶片线表数据乘以线速度,和俯仰角,得到叶片横截面的真实轮廓
  182. for df in result_line_tip:
  183. first_column = df.iloc[:, 0]
  184. sec_column = df.iloc[:, 1]
  185. df.iloc[:, 0] = first_column * v_speed_tip
  186. df.iloc[:, 1] = sec_column * np.cos(np.deg2rad(angle_tip_new))
  187. for df in result_line_root:
  188. first_column = df.iloc[:, 0]
  189. sec_column = df.iloc[:, 1]
  190. df.iloc[:, 0] = first_column * v_speed_root
  191. df.iloc[:, 1] = sec_column * np.cos(np.deg2rad(angle_root))
  192. for df in result_scatter_tip:
  193. first_column = df.iloc[:, 0]
  194. sec_column = df.iloc[:, 1]
  195. df.iloc[:, 0] = first_column * v_speed_tip
  196. df.iloc[:, 1] = sec_column * np.cos(np.deg2rad(angle_tip_new))
  197. for df in result_scatter_root:
  198. first_column = df.iloc[:, 0]
  199. sec_column = df.iloc[:, 1]
  200. df.iloc[:, 0] = first_column * v_speed_root
  201. df.iloc[:, 1] = sec_column * np.cos(np.deg2rad(angle_root))
  202. # 将叶片平均轮廓数据乘以线速度,得到实际叶片长度
  203. avg_tip = result_avg_tip.iloc[:, 0]
  204. result_avg_tip.iloc[:, 0] = avg_tip * v_speed_tip
  205. avg_root = result_avg_root.iloc[:, 0]
  206. result_avg_root.iloc[:, 0] = avg_root * v_speed_root
  207. twist_1 = round(np.abs(pitch_angle_root[0] - pitch_angle_tip[0]), 2)
  208. twist_2 = round(np.abs(pitch_angle_root[1] - pitch_angle_tip[1]), 2)
  209. twist_3 = round(np.abs(pitch_angle_root[2] - pitch_angle_tip[2]), 2)
  210. twist_avg = round((twist_1 + twist_2 + twist_3) / 3, 2)
  211. # 降低给数据采样频率,降低接口负担
  212. sampling_num = int(0.015 * sampling_fq_1)
  213. # 将原始数据的时间列由计时时钟转换为实际时间
  214. data_tip.iloc[:, 0] = data_tip.iloc[:, 0] / 5000000
  215. data_root.iloc[:, 0] = data_root.iloc[:, 0] / 5000000
  216. lowpass_data.iloc[:, 0] = lowpass_data.iloc[:, 0] / 5000000
  217. # 将需要保存到CSV的数据添加到return_list中
  218. return_list.append(str(time_code))
  219. return_list.append(str(wind_name))
  220. return_list.append(str(turbine_code))
  221. return_list.append(sampling_fq_1)
  222. return_list.append(pitch_angle_root[0])
  223. return_list.append(pitch_angle_root[1])
  224. return_list.append(pitch_angle_root[2])
  225. return_list.append(pitch_angle_root[3])
  226. return_list.append(mean_values[0])
  227. return_list.append(mean_values[1])
  228. return_list.append(mean_values[2])
  229. return_list.append(twist_1)
  230. return_list.append(twist_2)
  231. return_list.append(twist_3)
  232. return_list.append(twist_avg)
  233. return_list.append(tower_max)
  234. return_list.append(tower_freq)
  235. # 将return_list转换为DataFrame并追加到CSV文件
  236. df_new_row = pd.DataFrame([return_list],
  237. columns=['时间', '场站', '风机编号', '采样频率',
  238. '叶片1角度偏差', '叶片2角度偏差', '叶片3角度偏差', '相对角度偏差',
  239. '叶片1净空值', '叶片2净空值', '叶片3净空值',
  240. '叶片1扭转', '叶片2扭转', '叶片3扭转', '平均扭转',
  241. '振动幅值', '振动主频'])
  242. json_output = {
  243. 'original_plot': {
  244. 'blade_tip': {
  245. 'xdata': data_tip.iloc[:, 0].tolist()[::sampling_num],
  246. 'ydata': data_tip.iloc[:, 1].tolist()[::sampling_num]
  247. },
  248. 'blade_root': {
  249. 'xdata': data_root.iloc[:, 0].tolist()[::sampling_num],
  250. 'ydata': data_root.iloc[:, 1].tolist()[::sampling_num]
  251. }
  252. },
  253. 'fft_plot': {
  254. 'lowpass': {
  255. 'xdata': lowpass_data['time'].tolist()[::sampling_num],
  256. 'ydata': lowpass_data['distance_filtered'].tolist()[::sampling_num],
  257. 'xmax': max(lowpass_data['time'].tolist()),
  258. 'xmin': min(lowpass_data['time'].tolist()),
  259. 'ymax': max(lowpass_data['distance_filtered'].tolist()) + 0.02,
  260. 'ymin': min(lowpass_data['distance_filtered'].tolist()) - 0.02
  261. },
  262. 'fft': {
  263. 'xdata': fft_x,
  264. 'ydata': fft_y,
  265. 'xmax': max(fft_x),
  266. 'xmin': min(fft_x),
  267. 'ymax': max(fft_y) + 0.02,
  268. 'ymin': 0
  269. }
  270. },
  271. 'blade_tip': {
  272. 'first_blade': {
  273. 'xdata': result_line_tip[0].iloc[:, 0].tolist(),
  274. 'ydata': result_line_tip[0].iloc[:, 1].tolist()
  275. },
  276. 'second_blade': {
  277. 'xdata': result_line_tip[1].iloc[:, 0].tolist(),
  278. 'ydata': result_line_tip[1].iloc[:, 1].tolist()
  279. },
  280. 'third_blade': {
  281. 'xdata': result_line_tip[2].iloc[:, 0].tolist(),
  282. 'ydata': result_line_tip[2].iloc[:, 1].tolist()
  283. },
  284. 'avg_blade': {
  285. 'xdata': result_avg_tip.iloc[:, 0].tolist(),
  286. 'ydata': result_avg_tip.iloc[:, 1].tolist()
  287. }
  288. },
  289. 'blade_root': {
  290. 'first_blade': {
  291. 'xdata': result_line_root[0].iloc[:, 0].tolist(),
  292. 'ydata': result_line_root[0].iloc[:, 1].tolist()
  293. },
  294. 'second_blade': {
  295. 'xdata': result_line_root[1].iloc[:, 0].tolist(),
  296. 'ydata': result_line_root[1].iloc[:, 1].tolist()
  297. },
  298. 'third_blade': {
  299. 'xdata': result_line_root[2].iloc[:, 0].tolist(),
  300. 'ydata': result_line_root[2].iloc[:, 1].tolist()
  301. },
  302. 'avg_blade': {
  303. 'xdata': result_avg_root.iloc[:, 0].tolist(),
  304. 'ydata': result_avg_root.iloc[:, 1].tolist()
  305. }
  306. },
  307. 'dist_distribution': {
  308. 'first_blade': {
  309. 'xdata': dist_distribute[0].iloc[:, 0].tolist(),
  310. 'ydata': dist_distribute[0].iloc[:, 1].tolist()
  311. },
  312. 'second_blade': {
  313. 'xdata': dist_distribute[1].iloc[:, 0].tolist(),
  314. 'ydata': dist_distribute[1].iloc[:, 1].tolist()
  315. },
  316. 'third_blade': {
  317. 'xdata': dist_distribute[2].iloc[:, 0].tolist(),
  318. 'ydata': dist_distribute[2].iloc[:, 1].tolist()
  319. }
  320. },
  321. 'analyse_table': {
  322. 'pitch_angle_diff': {
  323. 'blade_1': pitch_angle_root[0],
  324. 'blade_2': pitch_angle_root[1],
  325. 'blade_3': pitch_angle_root[2],
  326. 'blade_relate': pitch_angle_root[3]
  327. },
  328. 'aero_dist': {
  329. 'first_blade': {
  330. 'x_min': min_keys[0],
  331. 'y_min': min_values[0],
  332. 'x_max': max_keys[0],
  333. 'y_max': max_values[0],
  334. 'y_diff': np.abs(max_values[0] - min_values[0]),
  335. 'y_ava': mean_values[0]
  336. },
  337. 'second_blade': {
  338. 'x_min': min_keys[1],
  339. 'y_min': min_values[1],
  340. 'x_max': max_keys[1],
  341. 'y_max': max_values[1],
  342. 'y_diff': np.abs(max_values[1] - min_values[1]),
  343. 'y_ava': mean_values[1]
  344. },
  345. 'third_blade': {
  346. 'x_min': min_keys[2],
  347. 'y_min': min_values[2],
  348. 'x_max': max_keys[2],
  349. 'y_max': max_values[2],
  350. 'y_diff': np.abs(max_values[2] - min_values[2]),
  351. 'y_ava': mean_values[2]
  352. }
  353. },
  354. 'blade_twist': {
  355. 'blade_1': twist_1,
  356. 'blade_2': twist_2,
  357. 'blade_3': twist_3,
  358. 'blade_avg': twist_avg
  359. },
  360. 'tower_vibration': {
  361. 'max_vibration': tower_max,
  362. 'main_vibration_freq': tower_freq
  363. }
  364. }
  365. }
  366. # 获取当前程序的绝对路径
  367. python_interpreter_path = sys.executable
  368. project_directory = os.path.dirname(python_interpreter_path)
  369. data_folder = os.path.join(project_directory, 'data')
  370. # 检查data文件夹是否存在,如果不存在则创建
  371. if not os.path.exists(data_folder):
  372. os.makedirs(data_folder)
  373. # CSV文件路径
  374. csv_file_path = os.path.join(data_folder, 'history_data.csv')
  375. # 检查CSV文件是否存在,如果不存在则创建一个空的CSV文件
  376. if not os.path.exists(csv_file_path):
  377. pd.DataFrame(columns=['时间', '场站', '风机编号', '采样频率',
  378. '叶片1角度偏差', '叶片2角度偏差', '叶片3角度偏差', '相对角度偏差',
  379. '叶片1净空值', '叶片2净空值', '叶片3净空值',
  380. '叶片1扭转', '叶片2扭转', '叶片3扭转', '平均扭转',
  381. '振动幅值', '振动主频']).to_csv(csv_file_path, index=False)
  382. df_new_row.to_csv(csv_file_path, mode='a', header=False, index=False)
  383. time_code_cleaned = time_code.replace("-", "").replace(":", "").replace(" ", "")
  384. json_filename = f"{wind_name}_{turbine_code}_{time_code_cleaned}.json"
  385. json_file_path = os.path.join(data_folder, json_filename)
  386. with open(json_file_path, 'w') as json_file:
  387. json.dump(json_output, json_file, indent=4)
  388. print('csv文件路径' + str(csv_file_path))
  389. print(result_line_tip[0].iloc[:, 0])
  390. print(result_line_root[0].iloc[:, 0])
  391. print('振动主频' + str(tower_freq))
  392. print('振动幅值' + str(tower_max))
  393. print('最小值', min_values)
  394. print('最小值对应的键', min_keys)
  395. print('最大值', max_values)
  396. print('最大值对应的键', max_keys)
  397. print('叶尖速度' + str(v_speed_tip), '叶根速度' + str(v_speed_root))
  398. print('新俯仰角' + str(angle_tip_new))
  399. print('轮毂中心距离' + str(dist_cen))
  400. print('叶根原始数据采样时间长度' + str(data_root.iloc[-1, 0]))
  401. # plot_data(result_line_tip, 'line', 'data1')
  402. # plot_data(result_diff_tip, 'line', 'data_diff_1')
  403. # plot_data(result_scatter_tip, 'scatter', 'data1')
  404. plot_data(result_line_root, 'line', 'data2')
  405. # plot_data(result_diff_root, 'line', 'data_diff_2')
  406. plot_data(result_scatter_root, 'scatter', 'data2')
  407. # plot_data(dist_distribute, 'scatter', 'dist_distribute')
  408. return json_output
  409. def process_data(file_path):
  410. """
  411. 打开、解决时间重置、按时间清洗异常值、分列数据
  412. """
  413. # 读取第2、4、9列的数据
  414. data = pd.read_csv(file_path, usecols=[1, 3, 8], header=None, engine='c')
  415. data = data.head(int(len(data) * 0.95))
  416. print('原始数据长度' + str(len(data)))
  417. '''
  418. # 绘制原始数据图
  419. # 只取前1%的数据
  420. # data = data.head(int(len(data)* 0.01))
  421. data.columns = ['time', 'distance1', 'distance2']
  422. plt.figure(figsize=(300, 150))
  423. sns.scatterplot(data=data, x='time', y='distance1', s=50, color='green')
  424. sns.scatterplot(data=data, x='time', y='distance2', s=50, color='red')
  425. abxy = plt.gca() # 获取当前坐标轴对象
  426. plt.grid(linewidth=2) # 设置网格线宽度为2
  427. abxy.xaxis.set_major_locator(MaxNLocator(nbins=100)) # 设置x轴主刻度的最大数量为10
  428. plt.xlabel('时间', fontsize=16, fontweight='bold') # 添加x轴标签
  429. plt.ylabel('距离(m)', fontsize=16, fontweight='bold') # 添加y轴标签
  430. abxy.tick_params(axis='x', labelsize=14, labelcolor='black', width=2) # 设置x轴刻度标签
  431. abxy.tick_params(axis='y', labelsize=14, labelcolor='black', width=2) # 设置y轴刻度标签
  432. plt.savefig(f"{"original"}.png", dpi=100, pil_kwargs={"icc_profile": False})
  433. plt.close()
  434. '''
  435. # 找到第一列中最大值和最小值的位置
  436. max_value = data.iloc[:, 0].max()
  437. max_index = data.iloc[:, 0].idxmax()
  438. min_index = data.iloc[:, 0].idxmin()
  439. # 检查最小值的位置是否是最大值位置的下一个
  440. if min_index == max_index + 1:
  441. # 将最小值及其之后的所有值都加上最大值
  442. data.iloc[min_index:, 0] += max_value
  443. # 按时间列筛选清洗异常值
  444. last_time = data.iloc[-1, 0]
  445. first_time = data.iloc[0, 0]
  446. filtered_data = data[(data.iloc[:, 0] > last_time) & (data.iloc[:, 0] < first_time)]
  447. print(f'时间列异常数据: {filtered_data}')
  448. print(f'起止时间: {first_time}, {last_time}')
  449. data = data[data.iloc[:, 0] >= first_time]
  450. data = data[data.iloc[:, 0] <= last_time]
  451. data.reset_index(drop=True, inplace=True)
  452. # 计算最小值
  453. min_time = data.iloc[:, 0].min()
  454. data.iloc[:, 0] -= min_time
  455. # 分为两组数据
  456. data_1 = data.iloc[:, [0, 1]]
  457. data_2 = data.iloc[:, [0, 2]]
  458. # 分别命名列
  459. data_1.columns = ['time', 'distance']
  460. data_2.columns = ['time', 'distance']
  461. return data_1, data_2
  462. def tower_filter(data_group: pd.DataFrame, noise_threshold: float):
  463. """
  464. 对轮毂中心数据进行降噪,和前项填充
  465. :param data_group: process_data计算完成后轮毂中心的数据。
  466. :param noise_threshold: 去掉占比小于noise_threshold的数据。
  467. :return: filtered_data:降噪后的数据
  468. """
  469. print('正在进行数据清洗......')
  470. time.sleep(1)
  471. # 计算distance的分布
  472. distance_counts = data_group['distance'].value_counts(normalize=True)
  473. noise_distance_threshold = distance_counts[distance_counts < noise_threshold].index
  474. noise_indices = data_group[data_group['distance'].isin(noise_distance_threshold)].index
  475. data_group.loc[noise_indices, 'distance'] = np.nan
  476. # 选择频率最大的5个值
  477. top_5_distances = distance_counts.head(5).index
  478. mean_values = data_group[data_group['distance'].isin(top_5_distances)]['distance'].mean()
  479. data_group.loc[(data_group['distance'] < mean_values-20) | (
  480. data_group['distance'] > mean_values*1.1), 'distance'] = np.nan
  481. nan_count = data_group['distance'].isna().sum()
  482. all_count = data_group.shape[0]
  483. print(f"中值是:{mean_values},替换为NaN的异常distance值的数量是: {nan_count}, 总数量是: {all_count},"
  484. f"占比: {nan_count / all_count * 100:.2f}%")
  485. # 前向填充
  486. data_group['distance'] = data_group['distance'].fillna(method='ffill')
  487. filtered_data = data_group
  488. return filtered_data
  489. def cycle_calculate(data_group: pd.DataFrame, noise_threshold: float, min_distance: float):
  490. """
  491. 对数据进行降噪,和前项填充;计算数据的周期节点,叶片前缘突变点、后缘突变点
  492. :param data_group: process_data计算完成后的数据。
  493. :param noise_threshold: 去掉占比小于noise_threshold的数据。
  494. :param min_distance: 区分叶片和塔筒的距离差值。
  495. :return: start_points:周期开始点, end_points:周期结束点, filtered_data:降噪后的数据
  496. """
  497. print('正在计算周期节点......')
  498. time.sleep(1)
  499. # 计算distance的分布
  500. distance_counts = data_group['distance'].value_counts(normalize=True)
  501. noise_distance_threshold = distance_counts[distance_counts < noise_threshold].index
  502. noise_indices = data_group[data_group['distance'].isin(noise_distance_threshold)].index
  503. data_group.loc[noise_indices, 'distance'] = np.nan
  504. # 选择频率最大的5个值
  505. top_5_distances = distance_counts.head(5).index
  506. mean_values = data_group[data_group['distance'].isin(top_5_distances)]['distance'].mean()
  507. data_group.loc[(data_group['distance'] < mean_values-30) | (
  508. data_group['distance'] > mean_values*1.1), 'distance'] = np.nan
  509. nan_count = data_group['distance'].isna().sum()
  510. all_count = data_group.shape[0]
  511. print(f"中值是:{mean_values},替换为NaN的distance异常值的数量是: {nan_count}, 总数量是: {all_count},"
  512. f"占比: {nan_count / all_count * 100:.2f}%")
  513. # 前向填充
  514. data_group['distance'] = data_group['distance'].fillna(method='ffill')
  515. filtered_data = data_group
  516. # 计算相邻两行distance的差值
  517. filtered_data['distance_diff'] = filtered_data['distance'].diff()
  518. large_diff_indices = filtered_data[filtered_data['distance_diff'] > min_distance].index
  519. small_diff_indices = filtered_data[filtered_data['distance_diff'] < -min_distance].index
  520. filtered_data = filtered_data.drop(columns=['distance_diff'])
  521. start_points = pd.DataFrame()
  522. end_points = pd.DataFrame()
  523. # 遍历所有差值大于的行
  524. for idx in large_diff_indices:
  525. # 获取当前行的 distance 值
  526. current_distance = filtered_data.loc[idx, 'distance']
  527. next_rows_large = filtered_data.loc[idx - 200: idx - 1]
  528. # 检查是否任意 distance 的值小于 current_distance - 2
  529. if next_rows_large['distance'].le(current_distance - min_distance).all():
  530. # 如果都小于,则将当前行和下一行添加到 special_points 中
  531. end_points = pd.concat([end_points, filtered_data.loc[[idx - 1]]])
  532. for idx in small_diff_indices:
  533. # 获取当前行的 distance 值
  534. current_distance = filtered_data.loc[idx - 1, 'distance']
  535. next_rows_small = filtered_data.iloc[idx: idx + 200]
  536. # 检查是否任意 distance 的值小于 current_distance - 2
  537. if next_rows_small['distance'].le(current_distance - min_distance).all():
  538. # 如果都小于,则将当前行和下一行添加到 special_points 中
  539. start_points = pd.concat([start_points, filtered_data.loc[[idx]]])
  540. if end_points.iloc[0, 0] < start_points.iloc[0, 0]:
  541. end_points = end_points.drop(end_points.index[0])
  542. if end_points.iloc[-1, 0] < start_points.iloc[-1, 0]:
  543. start_points = start_points.drop(start_points.index[-1])
  544. else:
  545. pass
  546. return start_points, end_points, filtered_data
  547. def data_normalize(data_group: pd.DataFrame, start_points: pd.DataFrame, end_points: pd.DataFrame) \
  548. -> Tuple[List[pd.DataFrame], List[pd.DataFrame], List[pd.DataFrame], int, list]:
  549. """
  550. 提取每个叶片的数据并归一化,输出散点图和拟合图
  551. :param data_group: cycle_calculate计算完成后的数据。
  552. :param start_points: 所有每个周期开始点,叶片前缘突变点。
  553. :param end_points: 叶片后缘突变点。
  554. :return: turbines_processed: 每个叶片的拟合数据,
  555. turbines_scattered: 每个叶片的散点数据,
  556. border_rows: 每个叶片的2个边缘数据,
  557. normalize_cycle: 周期长度
  558. """
  559. print('正在进行各周期归一化......')
  560. time.sleep(1)
  561. combined_df_sorted = pd.concat([start_points, end_points]).sort_values(by='time')
  562. # 检查排序后的数据从start开始,end结束
  563. if combined_df_sorted.iloc[0].equals(end_points.iloc[0]):
  564. combined_df_sorted = combined_df_sorted.iloc[1:]
  565. if combined_df_sorted.iloc[-1].equals(start_points.iloc[-1]):
  566. combined_df_sorted = combined_df_sorted.iloc[:-1]
  567. combined_df_sorted.reset_index(drop=True, inplace=True)
  568. # 将 start_points 中的时间点转换为列表
  569. start_times = combined_df_sorted['time'].tolist()
  570. print('本次测量风机完整旋转圈数:'+ str(len(start_times) / 2))
  571. time.sleep(1)
  572. normalize_cycle = start_times[1] - start_times[0]
  573. full_cycle = int((start_times[2] - start_times[0]) * 3)
  574. turbines = [pd.DataFrame() for _ in range(3)]
  575. # 遍历所有起始时间点
  576. for i in range(0, len(start_times), 2):
  577. # 获取当前起始和结束时间点
  578. start_time = start_times[i]
  579. end_time = start_times[i + 1]
  580. # 根据当前起始时间点和结束时间点对数据进行分段
  581. segment = data_group[(data_group['time'] > start_time) & (data_group['time'] <= end_time)]
  582. if segment is None:
  583. pass
  584. else:
  585. # 周期归一化
  586. ratio = (end_time - start_time) / normalize_cycle
  587. segment.loc[:, 'time'] = (segment['time'] - start_time) / ratio
  588. # segment.loc[:, 'distance'] = ff.butter_lowpass_filter(segment['distance'], cutoff_low, fs)
  589. # 将结果添加到相应的 turbine 数据框中
  590. turbines[i % 3] = pd.concat([turbines[i % 3], segment])
  591. # 数据分组清洗、求平均
  592. turbines_processed = []
  593. turbines_scattered = []
  594. min_list = []
  595. sd_time = [-1, -1]
  596. time_list = list(range(0, normalize_cycle, 9000))
  597. # time_list = [(i + 1) * normalize_cycle / fs * 100 for i in range(fs * 100)] # 生成时间序列
  598. for turbine in turbines:
  599. # 按时间排序
  600. turbine_sorted = turbine.sort_values(by='time').reset_index(drop=True)
  601. # 找到time列的第一个值
  602. first_time = turbine_sorted['time'].iloc[0]
  603. # 分组,时间列每1000为一组(每40个时间点一组)
  604. bins = list(range(int(first_time), int(turbine_sorted['time'].max()), 9000))
  605. # 原始代码
  606. # bins = list(range(int(first_time), int(turbine_sorted['time'].max()) + len(start_times), int(fs / 50)))
  607. grouped = turbine_sorted.groupby(pd.cut(turbine_sorted['time'], bins=bins, right=False))
  608. # 初始化一个空的 DataFrame 用于存储处理后的数据
  609. processed_df = pd.DataFrame()
  610. scattered_df = pd.DataFrame()
  611. mean_points = []
  612. diff_points = []
  613. # 对每个组进行处理
  614. for _, group in grouped:
  615. # 去除 distance 最大和最小的前5%
  616. quantile_5 = group['distance'].quantile(0.05)
  617. quantile_95 = group['distance'].quantile(0.95)
  618. filtered_group = group[(group['distance'] > quantile_5) & (group['distance'] < quantile_95)]
  619. # 计算均值
  620. mean_point = filtered_group['distance'].mean()
  621. mean_points.append(mean_point)
  622. # 遍历 mean_points 列表,计算每个元素与其下一个元素的差值
  623. for i in range(len(mean_points) - 1):
  624. diff = abs(mean_points[i + 1] - mean_points[i])
  625. diff_points.append(diff)
  626. start_index = int(len(diff_points) * 0.05)
  627. end_index = int(len(diff_points) * 0.95)
  628. subset1 = diff_points[start_index:end_index]
  629. sdr_diff = np.max(subset1) * 1.1
  630. min_list.append(min(mean_points))
  631. # 找到第一个和最后一个小于 sdr_diff 的序号
  632. first_index = np.where(diff_points < sdr_diff)[0][0]
  633. last_index = np.where(diff_points < sdr_diff)[0][-1]
  634. '''
  635. # 创建一个总图中有3个分图的形式
  636. fig, axs = plt.subplots(1, 1, figsize=(15, 9))
  637. plt.subplots_adjust(hspace=2)
  638. # 绘制 diff_points 的折线图
  639. axs.plot(diff_points, label='Diff Points', color='red', marker='x', markersize=5)
  640. axs.axhline(y=sdr_diff, color='red', linestyle='--')
  641. axs.legend()
  642. axs.set_title('Diff Points')
  643. axs.set_xlabel('Index')
  644. axs.set_ylabel('Value')
  645. # 显示图形
  646. plt.tight_layout()
  647. plt.show()
  648. '''
  649. for index, (bin, group) in enumerate(grouped):
  650. # 去除 distance 最大和最小的前5%
  651. quantile_5 = group['distance'].quantile(0.05)
  652. quantile_95 = group['distance'].quantile(0.95)
  653. filtered_group = group[(group['distance'] > quantile_5) & (group['distance'] < quantile_95)]
  654. if first_index <= index < last_index: # 如果斜率小于,则认为该组数据不是突变点
  655. # 计算中点
  656. mid_point = filtered_group.mean()
  657. # 将中点转换为 DataFrame 并添加到处理后的 DataFrame 中
  658. mid_point_df = pd.DataFrame([mid_point])
  659. mid_point_df.iloc[0, 0] = time_list[index]
  660. processed_df = pd.concat([processed_df, mid_point_df], ignore_index=True)
  661. scattered_df = pd.concat([scattered_df, filtered_group], ignore_index=True)
  662. else: pass
  663. # 找到time列的最小值和最大值
  664. min_time = processed_df['time'].min()
  665. max_time = processed_df['time'].max()
  666. if sd_time == [-1, -1]:
  667. sd_time = [min_time, max_time]
  668. elif sd_time[0] < min_time:
  669. sd_time[0] = min_time
  670. elif sd_time[1] > max_time:
  671. sd_time[1] = max_time
  672. # 将处理后的 DataFrame 添加到列表中
  673. turbines_processed.append(processed_df)
  674. turbines_scattered.append(scattered_df)
  675. # 把三组叶片数据按sd_time进行筛选,并把每个的边界数据保存
  676. border_rows = []
  677. for i, turbine in enumerate(turbines_processed):
  678. # 找到离 sd_time[0] 最近的行的索引
  679. closest_index_0 = (turbine['time'] - sd_time[0]).abs().idxmin()
  680. turbine.at[closest_index_0, 'time'] = sd_time[0]
  681. sd_time_row_0 = turbine.loc[closest_index_0]
  682. # 找到离 sd_time[1] 最近的行的索引
  683. closest_index_1 = (turbine['time'] - sd_time[1]).abs().idxmin()
  684. turbine.at[closest_index_1, 'time'] = sd_time[1]
  685. sd_time_row_1 = turbine.loc[closest_index_1]
  686. # 切片 turbine,从 closest_index_0 到 closest_index_1
  687. turbines_processed[i] = turbine.iloc[closest_index_0:closest_index_1 + 1].reset_index(drop=True)
  688. sd_time_rows_turbine = pd.concat([pd.DataFrame([sd_time_row_0]), pd.DataFrame([sd_time_row_1])]
  689. , ignore_index=True)
  690. border_rows.append(sd_time_rows_turbine)
  691. return turbines_processed, turbines_scattered, border_rows, full_cycle, min_list
  692. def blade_shape(turbines_processed: List[pd.DataFrame]):
  693. """
  694. 计算叶片平均形状、叶片形状偏差。
  695. :param turbines_processed:叶片拟合曲线数据,来自data_normalize
  696. :return: 叶片平均形状、叶片形状偏差
  697. """
  698. print('正在进行叶片外形偏差计算......')
  699. row_counts = [df.shape[0] for df in turbines_processed]
  700. num_rows = min(row_counts)
  701. # 创建一个新的data.frame用于保存结果
  702. turbine_avg = pd.DataFrame(index=range(num_rows), columns=['time', 'distance'])
  703. turbine_diff = [pd.DataFrame(index=range(num_rows), columns=['time', 'distance']) for _ in turbines_processed]
  704. # 遍历每一行
  705. for i in range(num_rows):
  706. distances = [df.loc[i, 'distance'] for df in turbines_processed] # 获取每个data.frame的distance列的值
  707. avg_distance = sum(distances) / len(distances) # 计算distance列的平均值
  708. time_value = turbines_processed[0].loc[i, 'time'] # 获取time列的值
  709. turbine_avg.loc[i, 'time'] = time_value
  710. turbine_avg.loc[i, 'distance'] = avg_distance
  711. for j in range(len(distances)):
  712. distances[j] = distances[j] - avg_distance
  713. turbine_diff[j].loc[i, 'time'] = time_value
  714. turbine_diff[j].loc[i, 'distance'] = distances[j]
  715. return turbine_avg, turbine_diff
  716. def coordinate_normalize(tip_border_rows: List[pd.DataFrame], tip_angle):
  717. """
  718. 将叶尖测量数据和叶根、轮毂中心的测量原点归一化。
  719. :param tip_border_rows: 3个叶尖边缘数据
  720. :param tip_angle: 叶尖测量俯仰角
  721. :return: 归一化后叶尖数据,叶尖俯仰角
  722. """
  723. tip_angle1 = np.deg2rad(tip_angle)
  724. tip_angle_list = []
  725. for turbine in tip_border_rows:
  726. tip_angle_cal0 = ((np.sin(tip_angle1) * turbine['distance'] - 0.07608) /
  727. (np.cos(tip_angle1) * turbine['distance']))
  728. tip_angle_cal = np.arctan(tip_angle_cal0)
  729. turbine['distance'] = (turbine['distance']**2 + 0.0057881664 -
  730. 0.15216*turbine['distance']*np.sin(tip_angle1)) ** 0.5
  731. tip_angle_list.append(tip_angle_cal)
  732. tip_angle_new = float(np.mean(tip_angle_list))
  733. tip_angle_new1 = np.rad2deg(tip_angle_new)
  734. print('坐标转换后的新叶尖俯仰角: ' + str(tip_angle_new1))
  735. return tip_border_rows, tip_angle_new1
  736. def radius_cal(border_rows, meas_angle, cen_dist, cen_angle, angle_main, angle_rotate):
  737. """
  738. 计算测量点处的旋转半径。
  739. :param border_rows: 三个叶片的边界
  740. :param meas_angle: 回波俯仰角
  741. :param cen_dist: 轮毂中心距离
  742. :param cen_angle: 轮毂中心俯仰角
  743. :param angle_main: 主轴倾角
  744. :param angle_rotate: 锥角
  745. :return: 旋转半径
  746. """
  747. aero_dist = (pd.concat([df['distance'] for df in border_rows]).mean())
  748. cen_x = np.cos(np.deg2rad(cen_angle)) * cen_dist
  749. cen_y = np.sin(np.deg2rad(cen_angle)) * cen_dist
  750. aero_x = np.cos(np.deg2rad(meas_angle)) * aero_dist
  751. aero_y = np.sin(np.deg2rad(meas_angle)) * aero_dist
  752. theta_4 = np.tan(np.pi - np.deg2rad(angle_main))
  753. theta_5 = np.tan(np.pi/2 - np.deg2rad(angle_main) - np.deg2rad(angle_rotate))
  754. if theta_5 > 1000:
  755. radius = np.abs((cen_y - aero_y) - theta_4 * (cen_x - aero_x))
  756. print("轴向倾角与锥角相近,叶片垂直于地面")
  757. else:
  758. radius = (np.abs((theta_4 * (cen_x - aero_x) - (cen_y - aero_y))/(theta_4 - theta_5))
  759. * ((1 + theta_5 ** 2) ** 0.5))
  760. print('测量点旋转半径:' + str(radius))
  761. return radius
  762. def blade_angle_aero_dist(border_rows: List[pd.DataFrame], radius: float, full_cycle: int,
  763. tower_dist: float, v_angle: float):
  764. """
  765. 计算叶片相对桨距角和叶片净空距离。
  766. :param border_rows: 三个叶片的边界
  767. :param radius: 旋转半径
  768. :param full_cycle: 全周期
  769. :param tower_dist: 塔筒距离
  770. :param v_angle: 俯仰角度
  771. :return: 绝对桨距角,净空距离,叶片线速度
  772. """
  773. print('正在进行相对桨距角和叶片净空距离计算......')
  774. v_speed = 2 * np.pi * radius / full_cycle # 叶片线速度m/(1计时器单位)
  775. pitch_angle_list = []
  776. aero_dist_list = []
  777. cen_blade = []
  778. for turbine in border_rows:
  779. diff_time = turbine.iloc[1, 0] - turbine.iloc[0, 0]
  780. diff_len = turbine.iloc[1, 1] - turbine.iloc[0, 1]
  781. mean_col2 = (turbine.iloc[1, 1] + turbine.iloc[0, 1]) / 2
  782. aero_dist = abs(mean_col2 - tower_dist) * np.cos(np.deg2rad(v_angle))
  783. pitch_angle = np.degrees(np.arctan(diff_len / (diff_time * v_speed)))
  784. print('单个叶片绝对桨距角' + str(pitch_angle))
  785. pitch_angle_list.append(pitch_angle)
  786. aero_dist_list.append(aero_dist)
  787. cen_blade.append(mean_col2)
  788. pitch_mean = np.mean(pitch_angle_list)
  789. pitch_angle_list = [angle - pitch_mean for angle in pitch_angle_list]
  790. pitch_angle_list.append(max(pitch_angle_list) - min(pitch_angle_list))
  791. aero_dist_list.append(np.mean(aero_dist_list))
  792. pitch_angle_list = [round(num, 2) for num in pitch_angle_list]
  793. aero_dist_list = [round(num, 2) for num in aero_dist_list]
  794. print('叶片相对角度偏差:' + '\n' + '叶片1:' + str(pitch_angle_list[0]) + '\n'
  795. + '叶片2:' + str(pitch_angle_list[1]) + '\n' + '叶片3:' + str(pitch_angle_list[2])
  796. + '\n' + '相对偏差范围:' + str(pitch_angle_list[3]))
  797. print('叶片净空距离:' + '\n' + '叶片1:' + str(aero_dist_list[0]) + '\n'
  798. + '叶片2:' + str(aero_dist_list[1]) + '\n' + '叶片3:' + str(aero_dist_list[2])
  799. + '\n' + '平均净空距离:' + str(aero_dist_list[3]))
  800. return pitch_angle_list, aero_dist_list, v_speed, cen_blade
  801. def plot_data(data, plot_type: str, data_name: str):
  802. """
  803. 绘制数据图表并保存为文件。
  804. :param data: 数据列表,每个元素是一个 DataFrame。
  805. :param plot_type: 图表类型,'line' 或 'scatter'。
  806. :param data_name: 数据名称,用于生成文件名。
  807. """
  808. print('正在画图......')
  809. time.sleep(1)
  810. save_path = "C:/Users/laiwe/Desktop/"
  811. save_name = fr"{data_name}_{plot_type}.png" # 生成文件名
  812. plt.figure(figsize=(300, 150))
  813. if plot_type == 'line':
  814. for df, color in zip(data, ['blue', 'green', 'red']):
  815. sns.lineplot(data=df, x=df.iloc[:, 0], y=df.iloc[:, 1], color=color)
  816. elif plot_type == 'scatter':
  817. for df, (size, color) in zip(data, [(50, 'blue'), (25, 'green'), (10, 'red')]):
  818. sns.scatterplot(data=df, x=df.iloc[:, 0], y=df.iloc[:, 1], s=size, color=color)
  819. else:
  820. raise ValueError("plot_type must be either 'line' or 'scatter'")
  821. axy = plt.gca() # 获取当前坐标轴对象
  822. plt.grid(which='both', linewidth=2) # 设置网格线宽度为2
  823. axy.xaxis.set_major_locator(MaxNLocator(nbins=200)) # 设置x轴主刻度的最大数量为10
  824. axy.yaxis.set_major_locator(MaxNLocator(nbins=100)) # 设置y轴主刻度的最大数量为10
  825. plt.xlabel('时间', fontsize=100, fontweight='bold') # 添加x轴标签
  826. plt.ylabel('距离(m)', fontsize=100, fontweight='bold') # 添加y轴标签
  827. axy.tick_params(axis='x', labelsize=10, labelcolor='black', width=2) # 设置x轴刻度标签
  828. axy.tick_params(axis='y', labelsize=60, labelcolor='black', width=10) # 设置y轴刻度标签
  829. plt.savefig(save_path + save_name)
  830. plt.close()
  831. abs_path = os.path.abspath(save_name)
  832. print(f" {save_name} 已完成")
  833. return abs_path
  834. def find_param(path: str):
  835. """
  836. 根据文件路径获取参数
  837. """
  838. path = path.replace('\\', '/')
  839. last_slash_index = path.rfind('/')
  840. result = path[last_slash_index + 1:]
  841. underscore_indices = []
  842. start = 0
  843. while True:
  844. index = result.find('_', start)
  845. if index == -1:
  846. break
  847. underscore_indices.append(index)
  848. start = index + 1
  849. wind_name = result[: underscore_indices[0]]
  850. turbine_code = result[underscore_indices[0] + 1: underscore_indices[1]]
  851. time_code = result[underscore_indices[1] + 1: underscore_indices[2]]
  852. sampling_fq = int(result[underscore_indices[2] + 1: underscore_indices[3]])
  853. tunnel_1 = float(result[underscore_indices[3] + 1: underscore_indices[4]])
  854. tunnel_2 = float(result[underscore_indices[4] + 1: -4])
  855. dt = datetime.strptime(time_code, "%Y%m%d%H%M%S")
  856. standard_time_str = dt.strftime("%Y-%m-%d %H:%M:%S")
  857. return wind_name, turbine_code, standard_time_str, sampling_fq, tunnel_1, tunnel_2
  858. def blade_dist_distribute_cal(data_group: pd.DataFrame, start_points: pd.DataFrame, end_points: pd.DataFrame,
  859. tower_dist: float, v_angle: float, blade_cen_dist: list):
  860. """
  861. 计算每个叶片每个周期的转速和净空距离
  862. :param data_group: cycle_calculate计算完成后的数据。
  863. :param start_points: 所有每个周期开始点,叶片前缘突变点。
  864. :param end_points: 叶片后缘突变点。
  865. :param tower_dist: 塔筒距离。
  866. :param v_angle: 测量俯仰角度。
  867. :param blade_cen_dist: 叶片内部距离。
  868. """
  869. print('正在进行各周期净空距离分布计算......')
  870. time.sleep(1)
  871. combined_df_sorted = pd.concat([start_points, end_points]).sort_values(by='time')
  872. # 检查排序后的数据从start开始,end结束
  873. if combined_df_sorted.iloc[0].equals(end_points.iloc[0]):
  874. combined_df_sorted = combined_df_sorted.iloc[1:]
  875. if combined_df_sorted.iloc[-1].equals(start_points.iloc[-1]):
  876. combined_df_sorted = combined_df_sorted.iloc[:-1]
  877. combined_df_sorted.reset_index(drop=True, inplace=True)
  878. # 将 start_points 中的时间点转换为列表
  879. start_times = combined_df_sorted['time'].tolist()
  880. normalize_cycle = start_times[1] - start_times[0]
  881. tower_clearance = [pd.DataFrame() for _ in range(3)]
  882. # 遍历所有起始时间点
  883. for i in range(0, len(start_times) - 2, 2):
  884. # 获取当前起始和结束时间点
  885. start_time = start_times[i]
  886. end_time = start_times[i + 1]
  887. # 根据当前起始时间点和结束时间点对数据进行分段
  888. segment = data_group[(data_group['time'] > start_time) & (data_group['time'] <= end_time)]
  889. min_distance = segment['distance'].min()
  890. clearance = np.abs(tower_dist - min_distance - blade_cen_dist[i % 3]) * np.cos(np.deg2rad(v_angle))
  891. r_speed = round(60 / ((start_times[i + 2] - start_times[i]) * 3 / 5000000), 2)
  892. new_df = pd.DataFrame({
  893. 'r_speed': [r_speed],
  894. 'clearance': [clearance]
  895. })
  896. # 将结果添加到相应的 turbine 数据框中
  897. tower_clearance[i % 3] = pd.concat([tower_clearance[i % 3], new_df])
  898. tower_clearance = [df.sort_values(by='r_speed') for df in tower_clearance]
  899. return tower_clearance
  900. # locate_path = "C:/Users/laiwe/Desktop/风电/激光测量/测试数据/20250728/gy_10-tf-20_20250630223600_20_13.03_23.32.csv"
  901. # measure_path= "C:/Users/laiwe/Desktop/风电/激光测量/测试数据/20250728/gy_10-tf-20_20250630223849_20_17.89_21.07.csv"
  902. locate_path = "C:/Users/laiwe/Desktop/风电/激光测量/测试数据/20250728/gy_10-tf-50_20250630223358_50_13.03_23.32.csv"
  903. measure_path= "C:/Users/laiwe/Desktop/风电/激光测量/测试数据/20250728/gy_10-tf-50_20250630224408_50_17.89_21.07.csv"
  904. # locate_path = "C:/Users/laiwe/Desktop/风电/激光测量/测试数据/20250728/gy_10-tf-100_20250630222752_100_13.03_23.32.csv"
  905. # measure_path= "C:/Users/laiwe/Desktop/风电/激光测量/测试数据/20250728/gy_10-tf-100_20250630225119_100_17.89_21.07.csv"
  906. # locate_path = "C:/Users/laiwe/Desktop/风电/激光测量/测试数据/20250728/gy_10-tff-20_20250630231223_20_12.51_20.06.csv"
  907. # measure_path= "C:/Users/laiwe/Desktop/风电/激光测量/测试数据/20250728/gy_10-tff-20_20250630232052_20_15.36_18.17.csv"
  908. # locate_path = "C:/Users/laiwe/Desktop/风电/激光测量/测试数据/20250728/gy_10-tff-50_20250630231417_50_12.51_20.06.csv"
  909. # measure_path= "C:/Users/laiwe/Desktop/风电/激光测量/测试数据/20250728/gy_10-tff-50_20250630233420_50_15.35_18.16.csv"
  910. # locate_path = "C:/Users/laiwe/Desktop/风电/激光测量/测试数据/20250728/gy_10-tff-100_20250630231610_100_12.51_20.06.csv"
  911. # measure_path= "C:/Users/laiwe/Desktop/风电/激光测量/测试数据/20250728/gy_10-tff-100_20250630234012_100_15.35_18.16.csv"
  912. start_t = time.time() # 记录开始时间
  913. data_path = [locate_path, measure_path, 5, 6]
  914. list_1 = data_analyse(data_path)
  915. # print(list_1)
  916. print(f"耗时: {time.time() - start_t:.2f} 秒")