numpy中where()函数理解

taojianglong

于 2020-08-09 16:44:31 发布

阅读量2k

点赞数 3

分类专栏： python numpy where 文章标签： python

本文链接：https://blog.youkuaiyun.com/qq_38915710/article/details/107895639

版权

np.where()有两种用法：

1. np.where(condition, x, y)

如果满足条件（condition）则输出x，不满足条件则输出y。

例子：

>>> import numpy as np
>>> a = np.arange(10)
>>> a
array([0, 1, 2, 3, 4, 5, 6, 7, 8, 9])
>>> np.where(a, 0, 1)
array([1, 0, 0, 0, 0, 0, 0, 0, 0, 0])
>>> np.where(a>2, 0, 1)
array([1, 1, 1, 0, 0, 0, 0, 0, 0, 0])
>>> np.where([[True, False], [True, True]], )
(array([0, 1, 1], dtype=int64), array([0, 0, 1], dtype=int64))
>>> np.where([[True, False], [True, True]], [[1, 3], [6, 4]], [[4, 2], [7, 8]])
array([[1, 2],
       [6, 4]])

np.where([[True, False], [True, True]], [[1, 3], [6, 4]], [[4, 2], [7, 8]])中[[True, False], [True, True]]为condition，第一个为True，输出1，第二个False，输出2，第三个True，输出6，第四个True，输出4

2. np.where(condition)

只有条件（conditio