引用几个例子
>>> a = np.array([-1, -2, 0, 1, 2])
>>> np.where(a)
(array([0, 1, 3, 4]),)
只有input,后面没有指定条件,返回非0的元素
>>>a = np.arange(10)
>>>a
array([0, 1, 2, 3, 4, 5, 6, 7, 8, 9]
本文详细介绍了Python中NumPy库的where函数,通过多个示例展示了其在条件判断和索引选取方面的应用。在没有指定条件时,where返回非零元素的索引;当设置条件时,它用于有条件地替换数组元素。此外,还解释了二维数组情况下where函数的工作原理,并讨论了返回两个一维数组的情况,以及如何根据这些索引理解元素位置。
引用几个例子
>>> a = np.array([-1, -2, 0, 1, 2])
>>> np.where(a)
(array([0, 1, 3, 4]),)
只有input,后面没有指定条件,返回非0的元素
>>>a = np.arange(10)
>>>a
array([0, 1, 2, 3, 4, 5, 6, 7, 8, 9]
2610
2万+

被折叠的 条评论
为什么被折叠?