Python：在matplotlib.pyplot插值的点处相交的图的交点

我正在寻找垂直线与我绘制的具有

pyplot

插值的绘图的交点

我想下面的代码和情节会让我的问题更清楚。下面是一些示例代码和结果图。我要寻找的是红色垂直线和蓝色线之间的所有交点（因此在这种情况下应该有3个这样的点）

我不知道怎么做——有人知道怎么做吗

守则：

import numpy as np
import matplotlib.pyplot as plt
%matplotlib inline

t = np.linspace(-np.pi, np.pi, 512, endpoint=False) + 0.0001 # 0.0001 to get rid of 0 values.

# normalized square wave
u = np.sign(np.sin(2 * np.pi * t))
u = u - np.min(u)
u = u / np.max(u)

# rotate the square wave
phi = - np.pi / 3.0
t_rot = t * np.cos(phi) - u * np.sin(phi)
u_rot = u * np.cos(phi) + t * np.sin(phi)

# level the rotated square wave
u_rot_leveled = u_rot + np.tan(-phi) * t_rot

plt.plot(t_rot, u_rot_leveled, '.-')
plt.axvline(x=-1.1, linestyle=':', color='red')

情节：

---

谢谢你的帮助

在

x==x0

的情况下，您可以实际找到

x-x0

相对于

y

的根（零），而不是插值

y

import numpy as np
import matplotlib.pyplot as plt

t = np.linspace(-np.pi, np.pi, 512, endpoint=False) + 0.0001 # 0.0001 to get rid of 0 values.

# normalized square wave
u = np.sign(np.sin(2 * np.pi * t))
u = u - np.min(u)
u = u / np.max(u)

# rotate the square wave
phi = - np.pi / 3.0
t_rot = t * np.cos(phi) - u * np.sin(phi)
u_rot = u * np.cos(phi) + t * np.sin(phi)

# level the rotated square wave
u_rot_leveled = u_rot + np.tan(-phi) * t_rot

def find_roots(x,y):
    s = np.abs(np.diff(np.sign(y))).astype(bool)
    return x[:-1][s] + np.diff(x)[s]/(np.abs(y[1:][s]/y[:-1][s])+1)

x0 = -1.1
z = find_roots(u_rot_leveled, t_rot-x0)

plt.plot(t_rot, u_rot_leveled, '.-')
plt.axvline(x=x0, linestyle=':', color='red')
plt.plot(np.ones_like(z)*x0, z, marker="o", ls="", ms=4, color="limegreen")

plt.show()

---

这里的部分解决方案来自我对

什么是新的？代码中没有定义它。一般来说，您可以使用

numpy.interp

来插值。噢，感谢您的关注。这应该是

u\u rot\u leveled

。我刚刚修复了这个错误和另一个错误，代码现在可以正常工作了。我尝试了几种不同的方法

numpy.interp

，但总是得到奇怪的结果，我不知道如何得到我想要的。