0
点赞
收藏
分享

微信扫一扫

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

49路末班车 2023-02-06 阅读 41


假设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


举报

相关推荐

0 条评论