import matplotlib.pyplot as plt
import numpy as np
from numpy.ma.core import where
a = np.array([10,5,6,7])
print(where(a==5))
print(a>6)
print(a[a>6])
print(a[np.logical_and(a>6,a<10)])
(array([1]),)
[ True False False True]
[10 7]
[7]
import matplotlib.pyplot as plt
import numpy as np
from numpy.ma.core import where
a = np.array([10,5,6,7])
print(where(a==5))
print(a>6)
print(a[a>6])
print(a[np.logical_and(a>6,a<10)])
(array([1]),)
[ True False False True]
[10 7]
[7]