【Python】np.nonzero()函数

参考：01_理解np.nonzero()函数 

for i in range(self.M.shape[0]):
    if len(np.nonzero(self.M[i, :])[1]) > self.anchor_thresh:
        candidate_anchors.append(i)
    print ("Number of candidates:", len(candidate_anchors))

跑GraphSTONE的时候，遇到了np.nonzero()函数，在此记录一下该函数的返回值。

np.nonzero(a)函数返回的是目标数组a中非零元素的索引，而len(np.nonzero(a))则表示目标数组a中非零元素的个数。

参考博客中举了三个例子来解释np.nonzero()的返回值，分别是一位数组、二维数组和三维数组，我们也依次实践一下。

1 一维数组

import numpy as np

a = [1, 0, 2, 0, 3, 0]   // 一维数组
b = np.nonzero(a)
print(b)

• 输出结果：

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

 np.nonzero()返回了一个一维array数组，其含义是数组中非零元素的索引，例子中的0,2,4即为数组a中非零元素1,2,3的索引。

---

2 二维数组

import numpy as np

a = [[1, 1, 0],
     [0, 0, 0],
     [1, 0, 3]]
b = np.nonzero(a)
print(b)

• 输出结果：

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

对于二维数组来说，np.nonzero()的返回值是两个array数组，对应二维数组的两个维度。[0][0]、[0][1]、[2][0]、[2][2]分别对应二维数组a中的非零元素，而返回的数组的长度则代表二维数组a中非零元素的个数。

---

3 三维数组

import numpy as np

a = [[[1, 1, 0],[0, 0, 1]],
     [[0, 0, 0],[1, 0, 0]],
     [[1, 0, 3],[0, 0, 0]]]
b = np.nonzero(a)
print(b)

• 输出结果

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

返回结果依旧是三维数组中非零元素的索引。