# Randomness Week 1 - Magnetic Pole Shift demo

import random

def pole_shift(n):
	x, y = 0, 0
	for i in range(n):
		(dx, dy) = random.choice([(0, 1), (0, -1), (1,0), (-1, 0)])
		x += dx
		y += dy
	return (x, y)


	number_of_walks = 5000

	for walk_length in range(1, 35):
		for i in range(number_of_walks):
			(x, y) = pole_shift(walk_length)
			distance = abs(x) + abs(y)