numpy where

版权声明:本文为作者创作,转载请注明出处:http://blog.csdn.net/claroja,如有商业用途请联系QQ:63183535。 https://blog.csdn.net/claroja/article/details/89155618
import numpy as np
a=np.array([0,2,4,6])
np.where(a>0)
Out[2]: (array([1, 2, 3], dtype=int64),)
b=np.array([[1,2,3],
            [4,5,6]])
np.where(b>3)
Out[3]: (array([1, 1, 1], dtype=int64), array([0, 1, 2], dtype=int64))

where返回的符合条件的元素的坐标

也可以直接修改值
condition : array_like, bool
Where True, yield x, otherwise yield y.

参考文献:
https://www.numpy.org/devdocs/reference/generated/numpy.where.html?highlight=where#numpy.where

展开阅读全文

没有更多推荐了,返回首页