numpy自学笔记

numpy中的条件筛选

# 利用np.where可以轻松 获得 符合条件的坐标

# 条件筛选
 #1.np.where 给定一个参数
a=np.arange(10).reshape(2,5)
print(a)
x,y=np.where(a>5)   # 返回一个元组 每个元素代表一维 每一个元素放着满足所有的满足此条件的所有的点的该维坐标
for  _x,_y in zip(x,y):
    print((_x,_y))
#  给定三个参数
print(np.where(a>5,1,0)) # 返回一个数组 将原数组中满足条件的元素赋值为1，否则时是0   # 例子中是1，0 可以是别的数字
#  2.通过 表达式 
print(a>1)   # a>1 返回一个数组 满足表达式时赋值为True，不满足赋值为False,[[False False  True  True  True]
 #                                                                [ True  True  True  True  True]]
# 利用这一特性可以筛选元素
print(a[a>1])
print(a[np.where(a>1)])    # 两者的效果是一样的
print(a)