numpy筛选指定的行或者列

n=np.arange (4*4)

n=n.reshape((4,4))

n=

array([[ 0,  1,  2,  3],
           [ 4,  5,  6,  7],
           [ 8,  9, 10, 11],
           [12, 13, 14, 15]])

假定我们要保留第3列小于9部分（绿色部分）

pos=np.where(n[:,2]<9)  # pos是满足条件的序号，是多维度的位置信息

n1=n[pos]  #直接使用pos

print(n1)

array([[0, 1, 2, 3],
       [4, 5, 6, 7]])

import numpy as np
n=np.arange (4*4)
n=n.reshape((4,4))
print(n)
pos=np.where(n[:,2]<9)
print("----筛选后------")
print(n[pos])