np.where()函数可以用来获取符合条件的值的索引.

例如,我们要获取矩阵中大于2的值的索引.

>>> a = np.array([1, 2, 3, 1, 2, 3, 1, 2, 3])
>>> idx = np.where(a > 2)
>>> idx
(array([2, 5, 8], dtype=int64),)

np.where()还有一种用法,直接来看代码

x = np.array([1, 2, 3, 4, 5])
y = np.array([11, 12, 13, 14, 15])
condition = np.array([True, False, True, True, False])
result = np.where(condition, x, y)

如果condition为True,那么result的值从x中取.
如果condition为False,那么result的值从y中取.

posted @ 2018-05-16 16:04:30
评论加载中...

发表评论