np.where()返回值的含义

np.where(condition)返回值是元祖。返回的是矩阵对应的位置,返回值需要用“行”和“列”表示。

以下图为例,原矩阵是个2维矩阵(5x5)。
第一行的[ 0, 0, 0, 0, 1, 1, 1, 1, 2, 2, 2, 3, 3, 3, 4, 4, 4]代表的是满足condition的行下标
第二行的[ 0, 1, 3, 4, 0, 2, 3, 4, 0, 2, 3, 0, 3, 4, 0, 1, 2]代表的是满足condition的列下标
合起来,就是(0,0),(0,1),(0,3).......

posted @ 2021-02-22 10:58  jungu  阅读(948)  评论(0)    收藏  举报