import math _SecPathTrans_LEN = 4 * 6 * 120 # _SecPathTransI_LEN = 4 * 6 * 120 _SpecRadius_LEN = 120 _PmTransferCoffs_LEN = 6 * 4 * 100 def tf_secPathTransData_process(sec_path_trans_data): _sec_path_trans_data = [] _sec_path_transR_data = sec_path_trans_data[0: _SecPathTrans_LEN] _sec_path_transI_data = sec_path_trans_data[_SecPathTrans_LEN: _SecPathTrans_LEN*2] for i in range(_SecPathTrans_LEN): _sec_path_trans_data.append((math.sqrt( math.pow(_sec_path_transR_data[i], 2) + math.pow(_sec_path_transI_data[i], 2) ))) return _sec_path_trans_data def tf_data_process(tf_data): _processed_tf_data = [] _processed_tf_data = tf_secPathTransData_process(tf_data) _offset = _SecPathTrans_LEN * 2 print('offset', _offset) _processed_tf_data +=tf_data[_offset:_offset + _SpecRadius_LEN + _PmTransferCoffs_LEN] return _processed_tf_data def tf_data_to_peaces(tf_data, len_mark_list): _processed_data = tf_data_process(tf_data) _peaces_data = [] _offset = 0 # 分割成片 for len_mark in len_mark_list: _peaces_data.append(_processed_data[_offset:_offset + len_mark]) _offset += len_mark return _peaces_data if __name__ == '__main__': _data = [] with open('../../Files/TF_dsp_data_20241124_144540.dat', 'r') as f: # tf_data = f.readlines() for line in f: num = float(line.strip().replace(',', '')) _data.append(num) print(len(_data)) # 对_data进行预处理 peaces_data = tf_data_to_peaces(_data, [ 120, 120, 120, 120, 120, 120, 120, 120, 120, 120, 120, 120, 120, 120, 120, 120, 120, 120, 120, 120, 120, 120, 120, 120, 120, 100, 100, 100, 100, 100, 100, 100, 100, 100, 100, 100, 100, 100, 100, 100, 100, 100, 100, 100, 100, 100, 100, 100, 100 ]) for peace in peaces_data: print(len(peace)) print(peace)