def emd_processing(input_folder, output_file): # 获取文件夹及其子文件夹中的所有 CSV 文件路径 csv_files = [] for root, dirs, files in os.walk(input_folder): for file in files: if file.endswith(".csv"): csv_files.append(os.path.join(root, file)) # 创建一个字典,用于存储每个列的处理结果 processed_data = {} # 遍历所有 CSV 文件 for file_path in csv_files: # 读取 CSV 文件 df = pd.read_csv(file_path) # 获取第一列数据 column_data = df.iloc[:, 0] # 对第一列进行 EMD 分解处理 emd = EMD() imfs = emd(column_data) # 将分解后的 IMF 数据存储到字典中 for i, imf in enumerate(imfs): new_column_name = f"IMF{i+1}" if new_column_name not in processed_data: processed_data[new_column_name] = [] processed_data[new_column_name].extend(imf) # 将处理后的数据写入新的 CSV 文件的每一列 df_output = pd.DataFrame(processed_data) df_output.to_csv(output_file, index=False) # 设置输入和输出路径 input_folder = currentPath output_file = currentPath # 调用 emd_processing 函数处理数据 emd_processing(input_folder, output_file)
时间: 2024-04-18 11:28:27 浏览: 147
论文研究 - 基于EMD和BP_AdaBoost神经网络的油价预测
感谢提供代码。根据你的代码,问题出现在 `emd()` 函数的调用上。
你在这里使用了 `emd = EMD()` 来创建一个 `EMD` 类的实例对象。然后,在下一行中,你尝试调用 `emd(column_data)` 来对 `column_data` 进行处理。
然而,根据你提供的代码,`emd()` 函数并没有在你的代码中定义。因此,当你尝试调用 `emd(column_data)` 时,会出现 `KeyError: -1` 的错误。
要解决这个问题,你需要确保在你的代码中定义了 `emd()` 函数或者从其他地方导入了该函数。如果 `emd()` 函数是从外部库导入的,你需要确保已经正确安装了该库,并在代码开头添加相应的导入语句。
请检查一下你的代码,并确保你定义了正确的 `emd()` 函数或者正确导入了该函数,这样就可以解决这个错误了。如果有更多问题,请随时提问。
阅读全文