numpy: np.where
Syntax
numpy.where(condition[, x, y])
返回 符合某一条件 的 下标 。
Note :
不接受 list 型的参数，只接受 `ndarray 型输入。
用法
单目运算
import numpy as np
array_1 = np.array([1, 0, 1, 5, 5, 1])
array_2 = np.array([[1, 0, 1],
[5, 5, 1]])
print np.where(array_1 > 2)
print np.where(array_2 > 2)打印结果：
(array([3, 4]),)
(array([1, 1]), array([0, 1]))三目运算
例：将奇数转换为偶数，偶数转换为奇数：
import numpy as np
y = np.array([1, 2, 3, 4, 5, 6])
print np.where(y%2 == 0, y+1, y-1)打印结果：
[0 3 2 5 4 7]本文参与 腾讯云自媒体同步曝光计划，分享自作者个人站点/博客。
原始发表：2017年11月23日，如有侵权请联系 cloudcommunity@tencent.com 删除
评论
登录后参与评论
推荐阅读
目录
腾讯云开发者
