import matplotlib.pyplot as plt
from matplotlib import style
import random

style.use('ggplot')


def generate_ys(hm,variance,correlation = False):
    val = 1
    ys = []
    for i in range(hm):
        y = val + random.randrange(-variance,variance)
        ys.append(y)
        if correlation and correlation == 'pos':
            val+=2
        elif correlation and correlation == 'neg':
            val-=2
    return ys




def create_dataset(hm,variance,correlation = False):
    val = 1
    ys = []
    for i in range(hm):
        y = val + random.randrange(-variance,variance)
        ys.append(y)
        if correlation and correlation == 'pos':
            val+=2
        elif correlation and correlation == 'neg':
            val-=2

    xs = [i for i in range(len(ys))]
    
    return xs,ys
        
xs, ys = create_dataset(40,40,correlation='pos')


plt.scatter(xs,ys,color='#003F72')
plt.show()