去掉数组中为0的元素

1.Python中去掉数组中的0元素

import numpy as np
#创建3D数组
a = np.array([[[1,0,1],[1,2,0],[2,3,3]],
             [[1,0,2],[1,2,3],[2,0,3]],
             [[1,0,6],[3,8,0],[6,4,3]]])
#去掉数组中重复元素
b = np.unique(a)
#>>>[0 1 2 3 4 6 8]
#去掉数组中零
b = b[b != 0]
#>>>[1 2 3 4 6 8]

2.matlab去掉数组中的0

a = [1,0,0;3,5,0;0,0,0];
a(find(a==0))=[]
%>>>[1,3,5]

3.matlab去掉数组中全0行（列）、含0行（列）

%去掉全零行
a(all(a==0,2),:)=[]
%>>>[1,0,0;3,5,0]
%去掉全零列
a(:,all(a==0,2))=[]
%>>[1,0;3,5;0,0]
%去掉含零列
a(:,any(a==0))=[]