演示

>>> import numpy
>>> A = numpy.array([[1,1],[2,[3,0],[4,[5,0]])
>>> A1 = A[A[:,2] == 1,:] # extract all rows with the third column 1
>>> A0 = A[A[:,2] == 0,:] # extract all rows with the third column 0
>>> A0
array([[3,0]])
>>> A1
array([[1,1]])