最近在看一些 Numpy 的知识, 看到知乎上一篇关于 where 的讲解特意转载一下
理解如下:
numpy.where()分两种调用方式:
1, 三个参数 np.where(cond,x,y): 满足 (cond) 输出 x , 不满足输出 y
2, 一个参数 np.where(array): 输出 arry 中 '真' 值的坐标(真也可以理解为非零)
1,np.where(cond,x,y):
同理:
np.where(x)输出的是八个不为 0 的数 (为'真'的数) 的坐标, 第一个 array[ ]是横坐标, 第一个 array[ ]是纵坐标
即: 如下图所示
来源: http://www.bubuko.com/infodetail-2901995.html