%matplotlib inline
import matplotlib.pyplot as plt
import numpy as np

nums = []
fin = open("lab0-numbers.txt")
for line in fin:
    nums.append(float(line.strip()))
plt.hist(nums)

import scipy.stats
scipy.stats.chisquare([  3.,   4.,   4.,   8.,   4.,   6.,  18.,   5.,   7.,   3.])

class RANDU():
    
    def __init__(self, s=1):
        self.SEED = s
        
    def random(self):
        self.SEED *= 65539
        self.SEED %= 2 ** 31
        return float(self.SEED) / 2 ** 31
    

r = RANDU()
for i in range(10):
    print r.random()

nums2 = []
for i in range(60):
    nums2.append(r.random())

plt.hist(nums2)

scipy.stats.chisquare([  7.,   1.,   2.,   9.,   6.,   5.,   6.,   6.,  11.,   7.])

x = []
y = []
z = []
for i in range(10000):
    x.append(r.random())
    y.append(r.random())
    z.append(r.random())

from mpl_toolkits.mplot3d import Axes3D

plt.scatter(y, z, s=1)

fig = plt.figure()
ax = fig.add_subplot(111, projection="3d")
ax.view_init(elev=20., azim=60)
ax.scatter(x, y, z, s=1)

x = []
y = []
z = []
for i in range(10000):
    x.append(np.random.random())
    y.append(np.random.random())
    z.append(np.random.random())

fig = plt.figure()
ax = fig.add_subplot(111, projection="3d")
ax.view_init(elev=20., azim=60)
ax.scatter(x, y, z, s=1)