Make mapColumnIndex support N dimensional columns and inputs