python get number of columns in numpy array