我发现没有关于如何反映对称 4d 数据的示例,这在想要使用对称平面执行 3d 模拟以减少计算(例如 ANSYS、COMSOL 等)时非常有用。此示例显示了对应于 COMSOL 仿真的数据文件结构,其结构为:X、Y、Z、振幅
该模型沿 Y 平面具有对称性,并在该平面上进行切片,因此必须计算较少的网格单元。为了获得完整的 Y 平面视图(即 Y 从 -0.5 到 0.5),数据必须沿 Y 平面反射。
此类问题的示例代码如下所示:
import numpy as np
import matplotlib.pyplot as plt
from scipy.interpolate import griddata
### generate a SIMULATION type-file###
X = np.linspace(-0.5, 0.5, 50)
Y = np.linspace(-0.5, 0, 50) #to be reflected/extended to 0.5
Z = np.linspace(-0.5, 0.5, 50)
Xq, Yq, Zq = np.meshgrid(X, Y, Z)
Amp = 1* np.exp(-((Xq - 0) ** 2 / (0.03) + ( Yq - 0) ** 2 / (0.03) + ( Zq - 0) ** 2 / (0.03)))
datafile = np.vstack([Xq.ravel(), Yq.ravel(), Zq.ravel(), Amp.ravel()]).T #resemble the simulation data structur, in this case X, Y, Z, Amp
### PYTHON POST-PROCESSING ###
X = datafile[:, 0]
Y = datafile[:, 1]
Z = datafile[:, 2]
Amp = datafile[:, 3] #Amplitude could be a Pressure, Force, Magnetic field, etc
xq = 0.0 #choose an arbitrary plane to show amplitude distribution over this plane
yq = np.linspace(min(Y), max(Y), 50)
zq = np.linspace(min(Z), max(Z), 50)
Xq, Yq, Zq = np.meshgrid(xq, yq, zq)
int_plane = griddata((X, Y, Z), Amp, (Xq, Yq, Zq), method='linear')
int_plane = np.squeeze(int_plane)
fig = plt.figure()
ax = fig.add_subplot(111, projection='3d')
ax.scatter(Zq, Yq, zs=int_plane)
ax.set_title('3D view');
ax.set_xlabel('x')
ax.set_ylabel('y')
ax.set_zlabel('z')
plt.show()
这里的目的是重建数据集的另一半(即整个其他 3D 半空间 + 相应的幅度)。3D 空间的正确反射将输出完整的 3D 高斯。如何有效地实现这一点?