numpy.where用法

Posted 2021-01-01 latup

函数原型：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都被指定

1 import numpy as np
2 
3 np.where([[False, False], [True, True]],    # codition: [[False, False], [True, True]]
4                [[1, 2], [3, 4]],            # x: [[1, 2], [3, 4]]
5                [[9, 8], [7, 6]])            # y: [[9, 8], [7, 6]]
6 
7 # 输出为：
8 # [[9 8]                                    # 对于codition中，位置(0, 0)和位置(0, 1)的值为False，所以该位置的输出对应于y的相应位置的值[9 8]
9 #  [3 4]]                                   # codition中，位置(1, 0)和位置(1, 1)的值为True，所以该位置的输出对应于x的相应位置的值[3 4]

 

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

1 import numpy as np
2 
3 print(np.where([[0, 1], [1, 1]]))         # 未指定x和y，只给出了condition，此时以元组的形式返回condition中非零元素的索引
4 
5 # 输出为:
6 # (array([0, 1, 1], dtype=int32),         # condition中，位置(0, 1), (1, 0), (1, 1)的值非零，对于轴0，非零值的索引为[0, 1, 1]
7 # 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