python中numpy.nonzero函数

numpy.nonzero函数

numpy.nonzero函数作用：返回非零元素的索引。

用法

import numpy as np

np.nonzero(a)

参数

• a：数组。

实例

实例1

x = np.array([[3, 0, 0], [0, 4, 0], [5, 6, 0]])
np.nonzero(x)

输出：

(array([0, 1, 2, 2], dtype=int64), array([0, 1, 0, 1], dtype=int64))

上述返回结果是一个包含2个元素的元组，每个元素都是数组类型，其中第一个元素中存放的是非零元素的行索引(0轴坐标，0轴表示垂直方向)，第二个元素存放的是非零元素的列索引(1轴坐标，1轴表示水平方向)。