Python numpy 2D array indexing,b[a].shape