热线电话:13121318867

登录
2018-10-30 阅读量: 1128
numpy where函数返回的是啥?

numpy的where函数用法:

#语法
where(condition, [x, y])

#condition :条件表达式
#x,y :可选参数,如果有x那么也要有y.
当没有x与y参数时,函数返回符合条件元素的坐标
当有x与y时,条件表达式为True的元素 返回x,条件表达式为False的元素返回y

例如

import numpy as np
a = np.array(range(27))
a
b = a.reshape(3,3,3)
b

当没有xy参数时:返回符合条件元素位置信息

np.where(b>5)

当有xy参数时,使用x替换符合条件的元素,使用y替换不符合条件的元素

np.where(b>5,-1,1)

0.0000
1
关注作者
收藏
评论(0)

发表评论

暂无数据
推荐帖子