python numpy库关键函数说明

np.argwhere()

输入参数是一个基本的逻辑表达式，输出检索结果的索引值。

>>> x = np.arange(6).reshape(2,3)
>>> x
array([[0, 1, 2],
       [3, 4, 5]])
>>> np.argwhere(x>1)
array([[0, 2],
       [1, 0],
       [1, 1],
       [1, 2]])

np.dtype()

用于实现自定义数据类型
链接: 详解dtype

np.shape()

用于获取矩阵的形状
链接: 详解shape

np.zeros()

用于生成元素是0的数