arr = [0,0,1,0,0] def swap(x, y): temp = x y = x x = temp def foo(arr): for i in range(len(arr)-1): if (arr[i] == 1): arr[i-1] = 1 arr[i+1] = 1 swap(arr[1], arr[2]) foo(arr) print(arr)