根据我们指定的条件检索函数中的元素
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()