python - Python中两个图的交集，找到x值

Question

让 0 <= x <= 1。我有两列 f 和 g 分别长度为 5000。现在我绘制:

plt.plot(x, f, '-')
plt.plot(x, g, '*')

我想找到曲线相交的点“x”。我不想找到 f 和 g 的交集。我可以简单地做到这一点:

set(f) & set(g)

Answer 1

您可以将 np.sign 与 np.diff 和 np.argwhere 结合使用来获取线交叉点的索引(在这种情况下，点是 [ 0, 149, 331, 448, 664, 743]):

import numpy as np
import matplotlib.pyplot as plt

x = np.arange(0, 1000)
f = np.arange(0, 1000)
g = np.sin(np.arange(0, 10, 0.01) * 2) * 1000

plt.plot(x, f, '-')
plt.plot(x, g, '-')

idx = np.argwhere(np.diff(np.sign(f - g))).flatten()
plt.plot(x[idx], f[idx], 'ro')
plt.show()

首先它使用 np.sign 计算 f - g 和相应的符号。应用 np.diff 会显示符号变化的所有位置(例如线条交叉)。使用 np.argwhere 为我们提供了准确的索引。