np.nonzero() 函数应用

作用 ： 返回非零值的索引

举例：

一维数组

得到一列索引值

data = np.array(list(range(9)))
idx = data.nonzero()
print(idx) # (array([1, 2, 3, 4, 5, 6, 7, 8]),)

二维数组

得到二列索引值 

data = np.array(list(range(9)))
data = data.reshape([3,3])
print(data)
[[0 1 2]
 [3 4 5]
 [6 7 8]]

idx1, idx2 = data.nonzero()
print( idx1, idx2 ) # [0 0 1 1 1 2 2 2] [1 2 0 1 2 0 1 2]
#两个数组分别对应 两个维度的索引，
#比如 地一个 元素 为  data[idx1[0], idx2[0]]

# 依次打印
for i in range(len(idx1)):
	print(data[idx1[i], idx2[i]])