快速找出Numpy中元素不为零的下标函数：np.nonzero() 快速建立主对角线Numpy数组函数：np.eye()

1.np.nonzero():

        找出数组中不为零元素下标。

例：

#### 10. Find indices of non-zero elements from [1,2,0,0,4,0] (★☆☆)

import numpy as np

nz = np.nonzero([1,2,0,0,4,0])

print(nz)

2.np.eye()

        一个参数：表明要创建一个几乘几的主对角线矩阵

import numpy as np

Z = np.eye(3)

print(Z)