plot中求线的交点

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()

f，g为y值，可以为不同维度