A close up of a camera lens with a circular design

$import numpy as np import matplotlib.pyplot as plt from mpl_toolkits.mplot3d import Axes3D # 参数设置 n = 1 pi = np.pi # 时空坐标计算 def spacetime_coordinates(t): r = n / pi * t x = r * np.cos(t) * np.cos(t) y = r * np.sin(t) * np.cos(t) z = r * np.sin(t) w = r return x, y, z, w # 量子场与概率 def quantum_field(t, m): u_n_minus_1 = 0 # 假设u_n-1为0，可根据实际情况修改 u = u_n_minus_1 + np.sin(t + m) return u # 概率分布 def probability_distribution(): m_values = [0, 1 / 2 * pi, pi, 3 / 2 * pi] probabilities = [0.125, 0.125, 0.125, 0.125, 0.5] m = np.random.choice(m_values + [0], p=probabilities) return m # 计算时空坐标和量子场值 t_values = np.linspace(0, 2 * pi, 100) coordinates = [spacetime_coordinates(t) for t in t_values] quantum_fields = [quantum_field(t, probability_distribution()) for t in t_values] # 绘制3D曲面 def plot_3d_surface(w_value): fig = plt.figure() ax = fig.add_subplot(111, projection='3d') x_values = [] y_values = [] z_values = [] u_values = [] for coord, field in zip(coordinates, quantum_fields): x, y, z, w = coord if w == w_value: x_values.append(x) y_values.append(y) z_values.append(z) u_values.append(field) ax.scatter(x_values, y_values, z_values, c=u_values, cmap='viridis') ax.set_xlabel('X') ax.set_ylabel('Y') ax.set_zlabel('Z') ax.set_title(f'3D Surface for w = {w_value}') plt.show() plot_3d_surface(1) plot_3d_surface(100) input("Press any key to exit") --auto$