根据我们指定的条件检索函数中的元素

import matplotlib.pyplot as plt
import numpy as np

a = np.linspace(0, 2 * np.pi, 50)
b = np.sin(a)
plt.plot(a,b)    #生成一个正弦函数图
mask = b >= 0
plt.plot(a[mask], b[mask], 'bo')  #符合条件的标注蓝色圆点
mask = (b >= 0) & (a <= np.pi / 2) 
plt.plot(a[mask], b[mask], 'go')  #符合条件的标注绿色圆点
plt.show()


import matplotlib.pyplot as plt
import numpy as np

a = np.linspace(0, 2 * np.pi, 50)
b = np.cos(a)
plt.plot(a,b)    #生成一个余弦函数图
mask = b >= 0
plt.plot(a[mask], b[mask], 'bo')  #符合条件的标注蓝色圆点
mask = (b >= 0) & (a <= np.pi / 2) 
plt.plot(a[mask], b[mask], 'go')  #符合条件的标注绿色圆点
plt.show()


import matplotlib.pyplot as plt
import numpy as np

a = np.linspace(0, 2 * np.pi, 50)
b = np.tan(a)
plt.plot(a,b)    #生成一个正切函数图
mask = b >= 0
plt.plot(a[mask], b[mask], 'bo')  #符合条件的标注蓝色圆点
mask = (b >= 0) & (a <= np.pi / 2) 
plt.plot(a[mask], b[mask], 'go')  #符合条件的标注绿色圆点
plt.show()

（"Drink from the well, replenish the well."--《The Walking Dead 》）