In [34] Instead of : # replace b[:,1,:] = [[9,9,9],[8,8]] We use : b[:,1,:] = [[9,9],[8,8]]