官方文档
两种用法
1.确定一个array中满足条件的索引列表
>>> import numpy as np
>>> a = np.arange(10)
>>> np.where(a < 5)
(array([0, 1, 2, 3, 4], dtype=int64),)
注意这里返回的是一个元组，需要获取到索引列表还需要[0]来访问返回结果的第一个元素
2.根据条件返回和原array形状一致的array
>>> import numpy as np
>>> a = np.arange(10)
>>> np.where(a < 5, a, 10*a)
array([ 0, 1, 2, 3, 4, 50, 60, 70, 80, 90])
三个参数：条件，条件为真的取值，条件为假的取值
直接返回一个array
版权声明：本文内容由互联网用户自发贡献，该文观点仅代表作者本人。本站仅提供信息存储空间服务，不拥有所有权，不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容， 请发送邮件至 举报，一经查实，本站将立刻删除。
文章由极客之音整理，本文链接：https://www.bmabk.com/index.php/post/133493.html
