Error ***index 1 is out of bounds for axis 0 with size 1('index 1 is out of bounds for axis 0 with size 1',)