Python 多维数组按某条件提取多个或单个维度数据

写代码的时候需要就做个记录~

先用布尔型看看

g=np.array([ [ 5, 12, -13],
             [ 3,  4,  5],
             [ 3,  4,  5]
                        ])
print(g[0]>0) #行提取
print(g[:,-1]>0) #列提取

按照某个条件提取

g=np.array([ [ 5, 12, -13],
             [ 3,  4,  5],
             [ 3,  4,  5]
                        ])
print(g[:,0]) #列提取
h=[g[:,i][g[:,-1]>0] for i in range(3)] #列提取