selecting specific column in each row from array,"array([0, 1, 0, 0, 0, 0, 0, 0, 0, 0])"