import numpy as np import pandas as pd import cv2 # 读取csv文件 df = pd.read_csv("3c_left_1-6.csv", header=None) data = df.values # 定义高斯滤波器函数 def gaussian_filter(data, sigma): # 计算高斯核 size = int(sigma * 3) if size % 2 == 0: size += 1 x, y, z = np.meshgrid(np.linspace(-1, 1, size), np.linspace(-1, 1, size), np.linspace(-1, 1, size)) kernel = np.exp(-(x ** 2 + y ** 2 + z ** 2) / (2 * sigma ** 2)) kernel /= kernel.sum() # 使用高斯核进行滤波 filtered_data = np.zeros_like(data) for i in range(data.shape[0]): filtered_data[i] = cv2.filter2D(data[i], -1, kernel, borderType=cv2.BORDER_REFLECT) return filtered_data # 对x、y、z方向上的时序信号分别进行高斯滤波 sigma = 1.5 # 高斯核标准差 filtered_data = np.zeros_like(data) for i in range(data.shape[1]): filtered_data[:, i] = gaussian_filter(data[:, i], sigma) for i in range(data.shape[0]): filtered_data[i] = gaussian_filter(filtered_data[i], sigma) for i in range(data.shape[2]): filtered_data[:, :, i] = gaussian_filter(filtered_data[:, :, i], sigma) # 计算SNR、MSE、PSNR snr = 10 * np.log10(np.sum(data**2) / np.sum((data-filtered_data)**2)) mse = np.mean((data - filtered_data) ** 2) psnr = 10 * np.log10(np.max(data)**2 / mse) print("SNR: {:.2f} dB".format(snr)) print("MSE: {:.2f}".format(mse)) print("PSNR: {:.2f} dB".format(psnr)) # 保存csv文件 df_filtered = pd.DataFrame(filtered_data) df_filtered.to_csv("filtered_data.csv", index=False, header=False)报错 File "G:\project2\gaussian.py", line 3, in <module> import cv2 ModuleNotFoundError: No module named 'cv2'

这个错误提示是因为你的Python环境中缺少了OpenCV库（cv2），你需要先安装OpenCV库才能运行这段代码。你可以在命令行中使用pip安装OpenCV库：

pip install opencv-python

或者使用Anaconda等Python集成环境来安装OpenCV库。安装完成后，重新运行代码即可。