我正在尝试执行简单的任务,例如在matplotlib中读取与y轴值对应的x轴值,我看不出有什么问题。
在这种情况下,我感兴趣的是例如找到y轴的哪个值,如果我选择x = 2.0,但我得idx
tuple为空,即使xvalues
数组中有数字2。
这是代码:
pyplot.plot(x,y,linestyle='--',linewidth=3)
ax = pyplot.gca()
line = ax.lines[0]
xvalues = line.get_xdata()
yvalues = line.get_ydata()
idx = where(xvalues == 2.0)
y = yvalues[idx[0][0]]
这是xvalues
数组:
[1.40000000e+00 1.45000000e+00 1.50000000e+00 1.55000000e+00
1.60000000e+00 1.65000000e+00 1.70000000e+00 1.75000000e+00
1.80000000e+00 1.85000000e+00 1.90000000e+00 1.95000000e+00
2.00000000e+00 2.05000000e+00 2.10000000e+00 2.15000000e+00
2.20000000e+00 2.25000000e+00 2.30000000e+00 2.35000000e+00]
答案 0 :(得分:8)
您获得空数组的原因是数组中实际上不存在严格值2.0
。
例如:
In [2]: x = np.arange(1.4, 2.4, 0.05)
In [3]: x
Out[3]:
array([ 1.4 , 1.45, 1.5 , 1.55, 1.6 , 1.65, 1.7 , 1.75, 1.8 ,
1.85, 1.9 , 1.95, 2. , 2.05, 2.1 , 2.15, 2.2 , 2.25,
2.3 , 2.35])
In [4]: x == 2.0
Out[4]:
array([False, False, False, False, False, False, False, False, False,
False, False, False, False, False, False, False, False, False,
False, False], dtype=bool)
In [5]: np.where(x == 2.0)
Out[5]: (array([], dtype=int64),)
这是浮点数学限制的经典问题。如果你愿意,你可以这样做:
y[np.isclose(x, 2)]
但是,通常,您希望在给定的x处插入y值。
例如,假设您想要2.01
处的值。 x值数组中不存在该值。
相反,使用np.interp
进行线性插值:
In [6]: y = np.cos(x)
In [7]: np.interp(2.01, x, y)
Out[7]: -0.4251320075130563