Numpy Count Number Of Columns In Matlab Which Function