np.where

作者: yalesaleng | 来源:发表于2018-07-16 15:40 被阅读19次
import numpy as np 

x = np.random.rand(9).reshape(3,3)
print('x:',x)
y = np.where(x>0.5)
print('y:',y)

# 输出结果
x: [[ 0.91075058  0.81927094  0.64425608]
    [ 0.10999029  0.19431304  0.20266795]
    [ 0.62419183  0.97548622  0.91133716]]
y: (array([0, 0, 0, 2, 2, 2], dtype=int64), array([0, 1, 2, 0, 1, 2], dtype=int64))

将输出的y化成一个2维矩阵来看,那么得到每一列都是满足条件x>0.5的像素的坐标,一共有6个像素满足条件,所以有6个坐标。

相关文章

网友评论

      本文标题:np.where

      本文链接:https://www.haomeiwen.com/subject/lcsbpftx.html