Python 曲线平移

Question

我想根据第 4 天给定的 P1 点将绿色曲线沿 Y 轴向上(或向下)移动。我想知道如何做到这一点，以便根据原始曲线(如果已移动)。

我知道绿色曲线的公式为:

0.000371 x^3 - 0.01274 x^2 + 0.1428 x - 0.02804

我需要解决两点:

1 - 如何根据 P1 移动绿色曲线。

2 - 将 P2 计算为绿色曲线移动的投影。

import numpy as np
import matplotlib.pyplot as plt

x = [1, 2, 3, 4, 5,6,7,8,9,10,11,12,13,14,15,16,17]
y = [0.1,0.2,0.3,0.38,0.41,0.46,0.475,0.478,0.486,0.489,0.493,0.49,0.495,0.498,0.511,0.5124,0.53]
plt.plot(x, y, 'x-')

# calculate polynomial
z = np.polyfit(x, y, 3)
f = np.poly1d(z)

# calculate new x's and y's
x_new = np.linspace(x[0], x[-1], 50)
y_new = f(x_new)

plt.plot(x,y,'o', label='original data')
plt.plot(x_new, y_new, label='fitted line')
plt.legend()
plt.show()

print(f)


# function to be defined :
p1 = 0.38 
p2 = findP2(f,p1) # to build
print(p2) 

希望有人能帮助我，谢谢!

:)

谢谢!

Answer 1

1. 要移动绿色曲线，您需要首先移动数据。从您的代码中，我猜测变量 p1 是点 P1 的 y 坐标。

import numpy as np
y = np.array(y) - y[3] + p1

p2 似乎只是 P2 x 坐标处的 polyfit 值。

如果此答案不正确，请给我更多信息。