用python实现储备池计算预测数据

import numpy as np import matplotlib.pyplot as pl from mpl_toolkits.mplot3d import Axes3D from scipy.optimize import fsolve def lorenz_s_e_befunc(u): global sigma, beta, rho, s, h, ub x, y, z, w = u xb, yb, zb, wb = ub return [ x - h*(sigma*(y-x)+s*w) - xb, y - h*(rho*x-y-x*z) - yb, z - h*(x*y-beta*z) - zb, w - h*(-x-sigma*w) - wb ] sigma, beta, rho, s = 1.0, 0.7, 26.0, 1.5 h = 0.001 #h = 0.07 n = 50000 result = np.zeros((n, 4)) ub = [1, 1, 1, 1] for i in range(n): y = fsolve(lorenz_s_e_befunc, [1, 1, 1, 1]) ub = y result[i] = y ax = Axes3D(pl.figure()) #ax.scatter(result[:, 0], result[:, 1], result[:, 2], s=1, marker='.') ax.plot(result[:, 0], result[:, 1], result[:, 2]) ax.set_xlabel('$x(t)$') ax.set_ylabel('$y(t)$') ax.set_zlabel('$z(t)$') pl.show()