python/numpy: how to get 2d array column length?,a.shape[1]