Selecting specific column in each row from array,"A[[0, 1], [0, 1]]"