去掉numpy数字中四周为0的行

功能很简单，但是代码让我折腾了一会，功能分享给大家！！！

import numpy as np


array1 = np.array([[1,0,1,0,0,0,0,0,0,1,1,0,0,0,1,1,0,1,0,0],
                   [0,1,1,0,0,1,1,1,1,0,0,0,1,0,1,0,0,1,1,1],
                   [0,0,1,0,0,1,1,1,0,0,0,0,0,0,0,1,0,0,1,1],
                   [0,1,1,0,0,1,1,1,1,0,1,1,1,0,0,1,0,0,1,1],
                   [0,0,1,0,0,1,1,1,0,1,0,1,1,0,1,1,0,0,1,0],
                   [1,0,1,0,0,0,1,0,0,1,1,1,1,0,1,1,0,0,1,0],
                   [1,0,1,0,1,1,0,0,0,0,1,0,0,0,1,0,0,0,1,1],
                   [0,1,0,0,1,0,0,0,1,0,1,1,1,0,1,0,0,1,1,0],
                   [0,1,0,0,1,0,0,1,1,0,1,1,1,0,0,1,0,1,0,0],
                   [1,0,0,0,0,1,0,1,0,0,0,1,1,0,0,1,0,1,0,0],
                   [1,0,0,0,0,1,0,1,0,0,0,1,1,0,0,1,0,1,0,0],
                   [0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0]])

print("raw  array", array1.shape)  #raw  array (12, 20)
mask = np.all((array1 == 0), axis=0)
column_indices = np.where(mask)[0]
array1 = array1[:,~mask]
mask1 = np.all((array1 == 0), axis=1)
array1 = array1[~mask1,:]

print("after array",array1.shape)  # after array (11, 17)
print("=====x=====\n",array1)

参考： https://blog.csdn.net/wsp_1138886114/article/details/108450410