自学教程：使用numpy查找元素位置numpy.where

numpy.where(condition,x,y)

import numpy as npa=np.array([1,2,3,4,5])print(np.where(a<3))#查找小于3的元素的位置

>>> x = np.arange(9.).reshape(3, 3)>>> np.where( x > 5 )(array([2, 2, 2]), array([0, 1, 2]))>>> x[np.where( x > 3.0 )] 
python实现矩阵的示例代码
python中numpy.dot()计算矩阵相乘