text: (cid:7) 
  
  # Set center ( column 1) and cue ( column 4) to give no reward observation ( row 1) 
  # Set reward and loss probabilities to 0.5 for the arm locations 
  
  # For reward condition right 
  A[2][:,:,1] = [ 1 .0 
  0 .0 
  0 .0  0 .0 
  0 .5 
  0 .5  0 .0 
  0 .5 
  0 .5  1 .0 
  0 .0 
  0 .0 ] 
  
  # For reward condition left 
  A[2][:,:,2] = [ 1 .0 
  0 .0 
  0 .0  0 .0 
  0 .5 
  0 .5  0 .0 
  0 .5 
  0 .5  1 .0 
  0 .0 
  0 .0 ] 
  (cid:6) 