import numpy as np N = 4 S = 10 a = 0.5 * np.ones(N) a[0] = 0.0 a[-1] = 1.0 print a for i in xrange(S): a[1:-1] = 0.5 * (a[:-2] + a[2:]) print a
Standard input is empty
[ 0. 0.5 0.5 1. ] [ 0. 0.25 0.75 1. ] [ 0. 0.375 0.625 1. ] [ 0. 0.3125 0.6875 1. ] [ 0. 0.34375 0.65625 1. ] [ 0. 0.328125 0.671875 1. ] [ 0. 0.3359375 0.6640625 1. ] [ 0. 0.33203125 0.66796875 1. ] [ 0. 0.33398438 0.66601562 1. ] [ 0. 0.33300781 0.66699219 1. ] [ 0. 0.33349609 0.66650391 1. ]