数学建模 - 离散拟合似连续

标签：return   from   ble   ima   lol   shape   lan   ret   range   

用离散值近似拟合连续值，可以用于快速理解题意。

import numpy as np
from matplotlib import pyplot as plt
import math

n = 240
x = np.zeros(shape=(4,n))
y = np.zeros(shape=(4,n))

x[0,0] = 100
y[0,0] = 0
x[1,0] = 0
y[1,0] = 0
x[2,0] = 0
y[2,0] = 100
x[3,0] = 100
y[3,0] = 100

def compute_direct(i,j):
    i2 = (i+1)%4
    cos_theta = (x[i2,j]-x[i,j]) / math.sqrt((x[i2,j]-x[i,j]) ** 2 + (y[i2,j]-y[i,j]) ** 2)
    sin_theta = (y[i2,j]-y[i,j]) / math.sqrt((x[i2,j]-x[i,j]) ** 2 + (y[i2,j]-y[i,j]) ** 2)
    return cos_theta, sin_theta


def move(i,j,v,dt):
    cos_theta, sin_theta = compute_direct(i,j)
    x[i,j+1] = x[i,j] + v * dt * cos_theta
    y[i,j+1] = y[i,j] + v * dt * sin_theta


dt = 0.05
v = 10

for _ in range(n-1):
    for i in range(4):
        move(i,_,v,dt)


plt.plot(x[0],y[0],‘red‘, label=‘A‘)
plt.plot(x[1],y[1],‘green‘, label=‘B‘)
plt.plot(x[2],y[2],‘blue‘, label=‘C‘)
plt.plot(x[3],y[3],‘black‘, label=‘D‘)
plt.legend()
plt.show()

数学建模 - 离散拟合似连续

标签：return   from   ble   ima   lol   shape   lan   ret   range   

原文地址：https://www.cnblogs.com/popodynasty/p/14969851.html