插值两个x - Python值的峰值

Question

我想在共振曲线的x图上找到两个y的值，它们与x-y的某个值相交。但是，由于我的数据点很少，我需要进行插值以找到x的这些值。

我正在看的曲线如下所示：

如何找到等于x值的y的两个值（以红色显示）？

我通过将数据拆分为两个数组来尝试np.interpolate：第一个使用gradient(y)>0而另一个使用gradient(y)<0，但这会产生不正确的值。然而，这种方法远非优雅，我寻求一个简单的解决方案。提前为您提供帮助。

其他信息： 到目前为止使用的代码：

from numpy import *
import pylab as pl
import numpy as np
import scipy as scipy
from scipy import optimize

#Get data
fn = '4K_peak_hiresGhz.txt'
F_values, S_values, P_values = loadtxt(fn, unpack=True, usecols=[1, 2, 3])

#Select Frequency range of peak
lower = 4.995
upper = 5.002
F_values_2 = F_values[(F_values>lower) & (F_values<upper)]
S_values_2 = S_values[(F_values>lower) & (F_values<upper)]
P_values_2 = P_values[(F_values>lower) & (F_values<upper)]

#Calculate Q-value of selected peak
S_Peak = max(S_values_2)
print('S21 peak (dB):')
print(S_Peak)
print

F_Peak = F_values_2[S_values_2.argmax()]
print('Freq at S21 peak (GHz)')
print(F_Peak)
print

width_S = S_Peak - 3
print('S21 peak - 3dB (dB)')
print(width_S)
print

f, axarr = pl.subplots(2, sharex=False)
axarr[0].set_xlabel('Frequency (GHz)')
axarr[0].plot(F_values_2,S_values_2)
axarr[0].plot(F_values_2,S_values_2,'.')
axarr[0].set_ylabel('S21 (dB)')
axarr[0].axhline(y=width_S, linewidth='0.7', ls='dashed',color='red')
axarr[0].axhline(y=S_Peak, linewidth='1', ls='dashed', color='black')
axarr[0].axvline(x=F_Peak, linewidth='1', ls='dashed', color='black')
axarr[1].scatter(F_values_2, P_values_2)
axarr[1].set_ylabel('Phase (deg)')
axarr[1].set_xlabel('Frequency (GHz)')
pl.show()

此外，此计划中分析的数据可在此dpaste中找到：http://dpaste.com/13AMJ92/

Answer 1

准备曲线数据：

from numpy import *
import pylab as pl
import numpy as np
import scipy as scipy
from scipy import optimize

#Get data
fn = '13AMJ92.txt'
F_values, S_values, P_values = loadtxt(fn, unpack=True, usecols=[1, 2, 3])

#Select Frequency range of peak
lower = 4.995
upper = 5.002
F_values_2 = F_values[(F_values>lower) & (F_values<upper)]
S_values_2 = S_values[(F_values>lower) & (F_values<upper)]

S_Peak = max(S_values_2)
F_Peak = F_values_2[S_values_2.argmax()]
width_S = S_Peak - 3

在峰值时拆分，并使用interp()：

idx = S_values_2.argmax()
x1 = np.interp(width_S, S_values_2[:idx+1], F_values_2[:idx+1])
x2 = np.interp(width_S, S_values_2[-1:idx-1:-1], F_values_2[-1:idx-1:-1])
print x1, x2

输出：

4.99902583799 4.99954573393

您也可以使用Shapely：

from shapely import geometry
curve = geometry.asLineString(np.c_[F_values_2, S_values_2])
hline = geometry.LineString([(F_values_2[0], width_S), (F_values_2[-1], width_S)])
print np.asarray(curve.intersection(hline))

输出：

[[  4.99902584 -21.958     ]
 [  4.99954573 -21.958     ]]

如果可以使用三次样条曲线，则可以使用InterpolatedUnivariateSpline：

from scipy import interpolate
us = interpolate.InterpolatedUnivariateSpline(F_values_2, S_values_2 - width_S)
x1, x2 = us.roots()
print x1, x2
pl.plot(F_values_2, S_values_2 - width_S)
x = np.linspace(F_values_2[0], F_values_2[-1], 100)
pl.plot(x, us(x))
pl.axhline(0)
pl.plot([x1, x2], [0, 0], "o")

输出：

4.99891956723 4.99960633369

情节：

Answer 2

在这种情况下，您可以使用np.interp()：

np.interp(width_S, S_values_2, F_values_2)

在你的情节中ticklabels与thickvalues不同，你可以比较一下这个事实：

print(axarr[0].get_xticks())
print([s.get_text() for s in axarr[0].get_xticklabels()])

使用此公式的插值是正确的，您必须设置正确的刻度值才能看到：

axarr[0].set_xticklabels(map(str, axarr[0].get_xticks()))