remove zero lines 2-D numpy array

前端 未结 3 1978
旧时难觅i
旧时难觅i 2020-12-01 04:07

I run a qr factorization in numpy which returns a list of ndarrays, namely Qand R:

>>&         


        
3条回答
  •  春和景丽
    2020-12-01 04:47

    Use np.all with an axis argument:

    >>> r[np.all(r == 0, axis=1)]
    array([[ 0.,  0.,  0.]])
    >>> r[~np.all(r == 0, axis=1)]
    array([[-1.41421356, -0.70710678, -0.70710678],
           [ 0.        , -1.22474487, -1.22474487]])
    

提交回复
热议问题