python怎么获取向量中非零元素的行号

假设A为向量  ：

行向量的时候,需要转化为列向量   A.T

A为列向量的时候,不需要转化

了解下numpy 中的nonzero()函数

np.nonzero(array or mmatrix)
    返回非零的行标，和列标

    [
      [1,2,0],
      [0,0,1],
      [3,0,0]]

    (array([0, 0, 1, 2], dtype=int64), array([0, 1, 2, 0], dtype=int64))
    行标的数组  [0, 0, 1, 2]  第0行两个元素   1行一个元素 2行一个元素
    列标的数组

A = np.mat([[1,2,3,0,5,6,8]]) #A为行向量

np.nonzero(A.T.A>0)[0]   #代表转化为array