#
#  Simple python random walk in 1D
#
import random
import numpy as np
import matplotlib.pyplot as plt
random.seed(None)        # Seed generator, None => system clock

def random_walk(steps):
    """  Performs 1D random walk with number steps = steps """
    walk = np.zeros(steps+1)
#######################################################
#   YOUR CODE GOES HERE: complete this function
#   Hint: use for loop to conduct all steps in
#        random walk.
#    Hint:  walk[i+1] = walk[i] + step ( + or -1)
#######################################################
    return walk

#  Perform random walk
num_steps = 1000
walk = random_walk(num_steps)
x = range(num_steps+1)
plt.plot(x,walk, 'b-')
plt.title('One-dimensional random walk')
plt.grid(True)
plt.show()