init_data.py

# Get the New York temperatures
n = 20
new_york_temp = city_temps.loc[city_temps['City'] == "New York"][-n:]
temps = np.array(new_york_temp["AverageTemperature"])
temps = temps / np.linalg.norm(temps)

# Extract some random training points from the 20 we are using (set random seed for reproducibility) 
training_points = np.atleast_2d([1, 5, 7, 10, 14, 16, 19]).T
np.random.seed(20)
train_temp = temps[training_points]