# whereDemo.py
import numpy as np
x = np.array([1,10,6,9,2,5,11,3])
print ('我们定义的一维数组是：')
print (x)
print ( '一维数组中大于 3 的元素的索引是：')
y = np.where(x >  3)  
print (y)
print ('使用这些索引来获取满足条件的元素：')
print (x[y])

x = np.arange(0,20,2).reshape(2,5)
print ('我们定义的二维数组是：')
print (x)
# 定义筛选条件
condition = np.logical_and(x > 10, x < 16)
print ( '二维数组中大于 10 且小于 16的元素的索引是：')
y = np.where(condition)
print (y)
print ('使用这些索引来获取满足条件的元素：')
print (x[y])

