How to crop a numpy 2d array to non-zero values?

后端 未结 3 1231
孤独总比滥情好
孤独总比滥情好 2020-12-11 18:48

Let\'s say i have a 2d boolean numpy array like this:

import numpy as np
a = np.array([
    [0,0,0,0,0,0],
    [0,1,0,1,0,0],
    [0,1,1,0,0,0],
    [0,0,0,0         


        
3条回答
  •  悲哀的现实
    2020-12-11 19:29

    a = np.transpose(a[np.sum(a,1) != 0])
    a = np.transpose(a[np.sum(a,1) != 0])
    

    It's not the quickest but it's alright.

提交回复
热议问题