python - 消除高于和低于某个值的地 block 值

Question

我正在绘制几个点，我想消除所有高于和低于给定范围的值。

所以我把我所有的点都画成

import numpy as np
import matplotlib.pylab as plt

我有两个数组:“magnitude”和“color”，以及最适合的方程，所以我绘制为:

plt.scatter(magnitude,color)
plt.plot(x,equation)
plt.plot(x,equation +0.5, linestyle =  '--')
plt.plot(x,equation-0.5, linestyle = '--')

其中 x 只是一些 linspace。它给了我一个看起来像这样的情节:

这正是我想要的，但现在我想删除红色虚线上方和下方的所有点(方程式 +0.5 和方程式 -0.5)，但我真的不知道该怎么做。提示？

Answer 1

您可以使用索引并创建一个定义边界条件的掩码，因为您的方程式在您执行算术运算时似乎是一个 NumPy 数组 (+0.5)作为

y1 = equation - 0.5
y2 = equation + 0.5

mask = (color>y1) & (color<y2)

plt.scatter(magnitude[mask], color[mask])
plt.plot(x, equation)
plt.plot(x, y1, linestyle =  '--')
plt.plot(x, y2, linestyle = '--')