numpy.where用法

函数原型：numpy.where(condition[, x, y])

 

参数介绍：condition：类似于矩阵，布尔值。当其值为True时，从x中选定值，否则，从y中选择值

　　　　   x，y：类似于矩阵，可选参数。根据condition的值来决定从x或y中取值

返回值：ndarray或者ndarrays的元组　

　　　　如果x和y都被指定，则输出矩阵的元素组成为，当某个位置的condition为True时， 该位置用相应的x值填充，若为False，则用相应的y值填充

　　　　如果只给出了condition，而未指定x和y，则返回元组condition.nonzero()　　　　　　　

　　　　（condition.nonzero()的作用是返回condition中值不为0的元素的索引，以元组的形式返回，元组由轴组成，每个轴的内容为该轴上值不为零的元素的索引）

情况1：x和y都被指定

import numpy as np

np.where([[False, False], [True, True]],    # codition: [[False, False], [True, True]]
               [[1, 2], [3, 4]],            # x: [[1, 2], [3, 4]]
               [[9, 8], [7, 6]])            # y: [[9, 8], [7, 6]]

# 输出为：
# [[9 8]                                    # 对于codition中，位置(0, 0)和位置(0, 1)的值为False，所以该位置的输出对应于y的相应位置的值[9 8]
#  [3 4]]                                   # codition中，位置(1, 0)和位置(1, 1)的值为True，所以该位置的输出对应于x的相应位置的值[3 4]

情况2：只给出condition，未指定x和y

import numpy as np

print(np.where([[0, 1], [1, 1]]))         # 未指定x和y，只给出了condition，此时以元组的形式返回condition中非零元素的索引

# 输出为:
# (array([0, 1, 1], dtype=int32),         # condition中，位置(0, 1), (1, 0), (1, 1)的值非零，对于轴0，非零值的索引为[0, 1, 1]
# array([1, 0, 1], dtype=int32))          # 对于轴1，非零值的索引为[1, 0, 1]，元组的元素由两个轴组成，所以为（[0, 1, 1], [1, 0, 1]）

文档链接：https://docs.scipy.org/doc/numpy/reference/generated/numpy.where.html