python numpy 2d array indexing,b[a].shape